codestyle

PaddlePaddle · Apr 17, 2023 · 66fbc73 · 66fbc73
1 parent fb55b07
commit 66fbc73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/phi/kernels/impl/dot_grad_kernel_impl.h b/paddle/phi/kernels/impl/dot_grad_kernel_impl.h
@@ -46,7 +46,7 @@ struct DotGradFunction<DeviceContext, T, phi::funcs::EnableComplex<T>> {
                   DenseTensor* tensor_dy) {
     VLOG(1) << "enable route";
 #if defined(__NVCC__) || defined(__HIPCC__)
-    if (1 >= tensor_dout->dims().size()) {
+    if (tensor_dout->dims().size() <= 1) {
       auto dout = EigenVector<T>::Flatten(*tensor_dout);
 
       if (tensor_dx) {