add defined cuda macro to cuda/conj_kernel.h

PaddlePaddle · Dec 20, 2021 · 8f4c482 · 8f4c482 · paddle-bot-old · Dec 20, 2021
1 parent 85dbe74
commit 8f4c482
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/paddle/pten/kernels/cuda/conj_kernel.cu b/paddle/pten/kernels/cuda/conj_kernel.cu
@@ -27,7 +27,6 @@ void Conj(const CUDAContext& dev_ctx, const DenseTensor& x, DenseTensor* out) {
 
 }  // namespace pten
 
-#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 PT_REGISTER_KERNEL(conj,
                    CUDA,
                    ALL_LAYOUT,
@@ -38,4 +37,3 @@ PT_REGISTER_KERNEL(conj,
                    double,
                    int,
                    int64_t) {}
-#endif
diff --git a/paddle/pten/kernels/cuda/conj_kernel.h b/paddle/pten/kernels/cuda/conj_kernel.h
@@ -14,6 +14,9 @@ limitations under the License. */
 
 #pragma once
 
+// CUDA and HIP use same api
+#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
+
 #include "paddle/pten/backends/cuda/cuda_context.h"
 #include "paddle/pten/core/dense_tensor.h"
 
@@ -23,3 +26,5 @@ template <typename T>
 void Conj(const CUDAContext& dev_ctx, const DenseTensor& x, DenseTensor* out);
 
 }  // namespace pten
+
+#endif