Skip to content

Commit c0f59da

Browse files
authored
【CUDA Kernel No.127-131】算子Kernel修复 (#2244)
1 parent 7c7ead9 commit c0f59da

9 files changed

Lines changed: 9 additions & 9 deletions

backends/iluvatar_gpu/kernels/cuda_kernels/moe_combine_no_weight_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
#include "paddle/phi/backends/gpu/gpu_context.h"
1616
#include "paddle/phi/core/dense_tensor.h"
1717
#include "paddle/phi/core/kernel_registry.h"
18-
#include "paddle/phi/kernels/legacy/gpu/moe_combine_no_weight_kernel.cu" // NOLINT
18+
#include "paddle/phi/kernels/legacy/gpu/moe_combine_no_weight_kernel.h"
1919

2020
PD_CUSTOM_KERNEL_REGISTER(moe_combine_no_weight,
2121
iluvatar_gpu,

backends/iluvatar_gpu/kernels/ernie_core/moe_combine_grad_kernel_register.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/moe_combine_grad_kernel.cu" // NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/moe_combine_grad_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(moe_combine_grad,
1818
iluvatar_gpu,

backends/iluvatar_gpu/kernels/ernie_core/moe_combine_kernel_register.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/moe_combine_kernel.cu" // NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/moe_combine_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(moe_combine,
1818
iluvatar_gpu,

backends/iluvatar_gpu/kernels/ernie_core/moe_gate_dispatch_kernel_register.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/moe_gate_dispatch_kernel.cu" // NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/moe_gate_dispatch_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(moe_gate_dispatch,
1818
iluvatar_gpu,

backends/metax_gpu/kernels/cuda_kernels/moe_combine_no_weight_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
#include "paddle/phi/backends/gpu/gpu_context.h"
1616
#include "paddle/phi/core/dense_tensor.h"
1717
#include "paddle/phi/core/kernel_registry.h"
18-
#include "paddle/phi/kernels/legacy/gpu/moe_combine_no_weight_kernel.cu" // NOLINT
18+
#include "paddle/phi/kernels/legacy/gpu/moe_combine_no_weight_kernel.h"
1919

2020
PD_CUSTOM_KERNEL_REGISTER(moe_combine_no_weight,
2121
metax_gpu,

backends/metax_gpu/kernels/ernie_core/moe_combine_grad_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/moe_combine_grad_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/moe_combine_grad_kernel.h"
1616
PD_CUSTOM_KERNEL_REGISTER(moe_combine_grad,
1717
metax_gpu,
1818
ALL_LAYOUT,

backends/metax_gpu/kernels/ernie_core/moe_combine_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/moe_combine_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/moe_combine_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(moe_combine,
1818
metax_gpu,

backends/metax_gpu/kernels/ernie_core/moe_gate_dispatch_grad_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/moe_gate_dispatch_grad_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/moe_gate_dispatch_grad_kernel.h"
1616
PD_CUSTOM_KERNEL_REGISTER(moe_gate_dispatch_grad,
1717
metax_gpu,
1818
ALL_LAYOUT,

backends/metax_gpu/kernels/ernie_core/moe_gate_dispatch_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414
#include "paddle/phi/core/kernel_registry.h"
15-
#include "paddle/phi/kernels/legacy/gpu/moe_gate_dispatch_kernel.cu" //NOLINT
15+
#include "paddle/phi/kernels/legacy/gpu/moe_gate_dispatch_kernel.h"
1616

1717
PD_CUSTOM_KERNEL_REGISTER(moe_gate_dispatch,
1818
metax_gpu,

0 commit comments

Comments
 (0)