fix compile error: only link dyload_cuda when cuda is available (Padd…

…lePaddle#50) * fix compile error: only link dyload_cuda when cuda is available
lijiaqi0612 · Sep 16, 2021 · 1e16889 · 1e16889
1 parent 0279d4a
commit 1e16889
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/paddle/fluid/operators/CMakeLists.txt b/paddle/fluid/operators/CMakeLists.txt
@@ -98,7 +98,12 @@ else()
     op_library(warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale)
 endif()
 
-op_library(spectral_op SRCS spectral_op.cc spectral_op.cu DEPS dynload_cuda ${OP_HEADER_DEPS})
+if (WITH_GPU AND (NOT WITH_ROCM))
+    op_library(spectral_op SRCS spectral_op.cc spectral_op.cu DEPS dynload_cuda ${OP_HEADER_DEPS})
+else()
+    op_library(spectral_op SRCS spectral_op.cc DEPS ${OP_HEADER_DEPS})
+endif()
+
 # op_library(spectral_op SRCS spectral_op.cc spectral_op.cu DEPS ${OP_HEADER_DEPS})
 # if (WITH_GPU)
 #     find_library(CUFFT_LIB libcufft.so