Skip to content

Commit aef8a3d

Browse files
committed
Standardize quantize_linear kernel and update CMakeLists
1 parent 71a6808 commit aef8a3d

3 files changed

Lines changed: 1 addition & 2 deletions

File tree

backends/iluvatar_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu

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

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/quantize_linear_kernel.cu" //NOLINT
1716
#include "paddle/phi/kernels/impl/quantize_linear_impl.h"
1817
#include "paddle/phi/kernels/quantize_linear_kernel.h"
1918

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -404,6 +404,7 @@ file(
404404
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/momentum_kernel.cu
405405
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/psroi_pool_grad_kernel.cu
406406
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/pixel_shuffle_grad_kernel.cu
407+
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/quantize_linear_kernel.cu
407408
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/lu_unpack_grad_kernel.cu
408409
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/nll_loss_grad_kernel.cu
409410
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/nanmedian_kernel.cu

backends/metax_gpu/kernels/cuda_kernels/quantize_linear_kernel_register.cu

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

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/quantize_linear_kernel.cu" //NOLINT
1716
#include "paddle/phi/kernels/impl/quantize_linear_impl.h"
1817
#include "paddle/phi/kernels/quantize_linear_kernel.h"
1918

0 commit comments

Comments
 (0)