fix

PaddlePaddle · Jan 3, 2024 · bb4d411 · bb4d411
1 parent 8ebf9a2
commit bb4d411
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 29 deletions.
diff --git a/paddle/phi/kernels/gpu/nuclear_norm_grad_kernel.cu b/paddle/phi/kernels/gpu/nuclear_norm_grad_kernel.cu
@@ -24,6 +24,4 @@ PD_REGISTER_KERNEL(nuclear_norm_grad,
                    ALL_LAYOUT,
                    phi::NuclearNormGradKernel,
                    float,
-                   double,
-                   phi::dtype::float16,
-                   phi::dtype::bfloat16) {}
+                   double) {}
diff --git a/paddle/phi/kernels/gpu/nuclear_norm_kernel.cu b/paddle/phi/kernels/gpu/nuclear_norm_kernel.cu
@@ -19,11 +19,5 @@
 
 #include "paddle/phi/kernels/impl/nuclear_norm_kernel_impl.h"
 
-PD_REGISTER_KERNEL(nuclear_norm,
-                   GPU,
-                   ALL_LAYOUT,
-                   phi::NuclearNormKernel,
-                   float,
-                   double,
-                   phi::dtype::float16,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    nuclear_norm, GPU, ALL_LAYOUT, phi::NuclearNormKernel, float, double) {}
diff --git a/paddle/phi/kernels/gpu/p_norm_grad_kernel.cu b/paddle/phi/kernels/gpu/p_norm_grad_kernel.cu
@@ -104,11 +104,5 @@ void PNormGradKernel(const Context& dev_ctx,
   }
 }
 }  // namespace phi
-PD_REGISTER_KERNEL(p_norm_grad,
-                   GPU,
-                   ALL_LAYOUT,
-                   phi::PNormGradKernel,
-                   float,
-                   double,
-                   phi::dtype::float16,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(
+    p_norm_grad, GPU, ALL_LAYOUT, phi::PNormGradKernel, float, double) {}
diff --git a/paddle/phi/kernels/gpu/p_norm_kernel.cu b/paddle/phi/kernels/gpu/p_norm_kernel.cu
@@ -96,11 +96,4 @@ void PNormKernel(const Context& dev_ctx,
 }
 }  // namespace phi
 
-PD_REGISTER_KERNEL(p_norm,
-                   GPU,
-                   ALL_LAYOUT,
-                   phi::PNormKernel,
-                   float,
-                   double,
-                   phi::dtype::float16,
-                   phi::dtype::bfloat16) {}
+PD_REGISTER_KERNEL(p_norm, GPU, ALL_LAYOUT, phi::PNormKernel, float, double) {}
diff --git a/paddle/phi/kernels/impl/nuclear_norm_kernel_impl.h b/paddle/phi/kernels/impl/nuclear_norm_kernel_impl.h
@@ -99,7 +99,5 @@ void NuclearNormKernel(const Context& dev_ctx,
     phi::SumKernel<T, Context>(
         dev_ctx, singular_tensor, {-1}, singular_tensor.dtype(), keepdim, out);
   }
-
-  std::cout << "sum_grad.dims():" << singular_tensor.dims() << std::endl;
 }
 }  // namespace phi