We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0279d4a commit 1e16889Copy full SHA for 1e16889
paddle/fluid/operators/CMakeLists.txt
@@ -98,7 +98,12 @@ else()
98
op_library(warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale)
99
endif()
100
101
-op_library(spectral_op SRCS spectral_op.cc spectral_op.cu DEPS dynload_cuda ${OP_HEADER_DEPS})
+if (WITH_GPU AND (NOT WITH_ROCM))
102
+ op_library(spectral_op SRCS spectral_op.cc spectral_op.cu DEPS dynload_cuda ${OP_HEADER_DEPS})
103
+else()
104
+ op_library(spectral_op SRCS spectral_op.cc DEPS ${OP_HEADER_DEPS})
105
+endif()
106
+
107
# op_library(spectral_op SRCS spectral_op.cc spectral_op.cu DEPS ${OP_HEADER_DEPS})
108
# if (WITH_GPU)
109
# find_library(CUFFT_LIB libcufft.so
0 commit comments