调整代码格式

PaddlePaddle · Apr 7, 2022 · 3e2e62a · 3e2e62a
1 parent 7b9c8ed
commit 3e2e62a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 14 deletions.
diff --git a/paddle/phi/kernels/cpu/logspace_kernel.cc b/paddle/phi/kernels/cpu/logspace_kernel.cc
@@ -13,10 +13,11 @@
 // limitations under the License.
 
 #include <cmath>
-#include "paddle/phi/kernels/logspace_kernel.h"
+
 #include "paddle/phi/backends/cpu/cpu_context.h"
 #include "paddle/phi/core/kernel_registry.h"
 #include "paddle/phi/kernels/funcs/data_type_transform.h"
+#include "paddle/phi/kernels/logspace_kernel.h"
 
 namespace phi {
 
@@ -52,11 +53,11 @@ void LogspaceKernel(const Context& ctx,
     int half_num = num / 2;
     for (int i = 0; i < num; ++i) {
       if (i < half_num) {
-        out_data[i] = static_cast<T>(std::pow(
-          base_data, start_data + step * i));
+        out_data[i] =
+            static_cast<T>(std::pow(base_data, start_data + step * i));
       } else {
-        out_data[i] = static_cast<T>(std::pow(
-          base_data, stop_data - step * (num - i - 1)));
+        out_data[i] = static_cast<T>(
+            std::pow(base_data, stop_data - step * (num - i - 1)));
       }
     }
   } else {

diff --git a/paddle/phi/kernels/gpu/logspace_kernel.cu b/paddle/phi/kernels/gpu/logspace_kernel.cu
@@ -29,21 +29,21 @@ __global__ void LogspaceKernelInner(
 
   for (; index < size; index += blockDim.x * gridDim.x) {
     if (index < size / 2) {
-      out[index] = static_cast<T>(pow(
-          static_cast<double>(base),
-          static_cast<double>(start + step * index)));
+      out[index] =
+          static_cast<T>(pow(static_cast<double>(base),
+                             static_cast<double>(start + step * index)));
     } else {
-      out[index] = static_cast<T>(pow(
-          static_cast<double>(base),
-          static_cast<double>(stop - step * (size - index - 1))));
+      out[index] = static_cast<T>(
+          pow(static_cast<double>(base),
+              static_cast<double>(stop - step * (size - index - 1))));
     }
   }
 }
 
 template <typename T>
 __global__ void LogspaceSpecialKernel(T start, T base, T* out) {
-  out[0] = static_cast<T>(pow(
-      static_cast<double>(base), static_cast<double>(start)));
+  out[0] = static_cast<T>(
+      pow(static_cast<double>(base), static_cast<double>(start)));
 }
 
 template <typename T, typename Context>

diff --git a/python/paddle/fluid/tests/unittests/test_logspace.py b/python/paddle/fluid/tests/unittests/test_logspace.py
@@ -222,4 +222,4 @@ def test_base_dtype():
 
 
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()