mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
249614ca88
Summary: Closes https://github.com/caffe2/caffe2/pull/767 Differential Revision: D5292618 Pulled By: akyrola fbshipit-source-id: 22bcfe01244d6beb48c580c84c790c810dc06998
12 lines
386 B
CMake
12 lines
386 B
CMake
if(USE_CUDA)
|
|
set(Caffe2_CUDA_RTC_GPU_SRC
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/elemenntwise_rtc_gpu.cc"
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/pool_op_rtc_gpu.cc"
|
|
)
|
|
|
|
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${Caffe2_CUDA_RTC_GPU_SRC})
|
|
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} PARENT_SCOPE)
|
|
else()
|
|
message(STATUS "CUDA RTC operators skipped due to no CUDA support")
|
|
endif()
|