Skip to content

Commit 9179143

Browse files
SpongeBob0318luotao1YqGe585
authored
CUDA Kernel No.14】fused_transpose_wlch_split_quant算子Kernel修复 (PaddlePaddle#2012)
Co-authored-by: Tao Luo <luotao02@baidu.com> Co-authored-by: Yuqiang Ge <143453447+YqGe585@users.noreply.github.com>
1 parent de0ad95 commit 9179143

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

backends/iluvatar_gpu/kernels/cuda_kernels/fused_transpose_wlch_split_quant_kernel_register.cu

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/fusion/gpu/fused_transpose_wlch_split_quant_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/fusion/gpu/fused_transpose_wlch_split_quant_kernel.h"
1717
#include "paddle/phi/kernels/fusion/gpu/quant_utils.h"
1818
#include "paddle/phi/kernels/primitive/datamover_primitives.h"
1919

backends/metax_gpu/kernels/fusion/fused_transpose_wlch_split_quant_kernel_register.cu

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/fusion/gpu/fused_transpose_wlch_split_quant_kernel.cu" //NOLINT
16+
#include "paddle/phi/kernels/fusion/gpu/fused_transpose_wlch_split_quant_kernel.h"
1717
#include "paddle/phi/kernels/fusion/gpu/quant_utils.h"
1818
#include "paddle/phi/kernels/primitive/datamover_primitives.h"
1919

0 commit comments

Comments
 (0)