From 795418750e231917284089a51e1db18789ebfc05 Mon Sep 17 00:00:00 2001 From: co63oc Date: Mon, 2 Sep 2024 16:20:50 +0800 Subject: [PATCH] Fix --- paddle/phi/kernels/gpu/matmul_grad_kernel.cu | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/paddle/phi/kernels/gpu/matmul_grad_kernel.cu b/paddle/phi/kernels/gpu/matmul_grad_kernel.cu index ef3ce2cabf3858..72ca06bad3d5aa 100644 --- a/paddle/phi/kernels/gpu/matmul_grad_kernel.cu +++ b/paddle/phi/kernels/gpu/matmul_grad_kernel.cu @@ -71,9 +71,7 @@ PD_REGISTER_KERNEL(legacy_matmul_grad, GPU, ALL_LAYOUT, phi::LegacyMatmulGradKernel, + int8_t, float, double, - phi::dtype::float16, - phi::dtype::bfloat16, - phi::dtype::complex, - phi::dtype::complex) {} + phi::dtype::float16) {}