Fix

PaddlePaddle · Sep 2, 2024 · 7954187 · 7954187
1 parent cc4fcdc
commit 7954187
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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<float>,
-                   phi::dtype::complex<double>) {}
+                   phi::dtype::float16) {}