Skip to content

Commit 9a27d17

Browse files
Le-soleileYqGe585
andauthored
【CUDA Kernel No.105】set_value算子Kernel修复 (PaddlePaddle#2018)
* Change include * For ci --------- Co-authored-by: Yuqiang Ge <[email protected]>
1 parent 9179143 commit 9a27d17

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/set_value_kernel_register.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
// limitations under the License.
1414

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/set_value_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/set_value_kernel.h"
1717

1818
PD_CUSTOM_KERNEL_REGISTER(set_value,
1919
iluvatar_gpu,

backends/metax_gpu/kernels/cuda_kernels/set_value_kernel_register.cu

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,10 @@
1212
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1313
// See the License for the specific language governing permissions and
1414
// limitations under the License.
15-
// clang-format off
15+
1616
#include "paddle/phi/core/kernel_registry.h"
17-
#include "paddle/phi/kernels/gpu/set_value_kernel.cu" // NOLINT
1817
#include "paddle/phi/kernels/set_value_kernel.h"
19-
// clang-format on
18+
2019
PD_CUSTOM_KERNEL_REGISTER(set_value,
2120
metax_gpu,
2221
ALL_LAYOUT,

0 commit comments

Comments
 (0)