Commit fe0f99b2 by Tatsuya Nishiyama Committed by Tianqi Chen

Fix cmake for building with cuda in msvc (#1437)

parent 25db77f8
...@@ -29,8 +29,8 @@ macro(find_cuda use_cuda) ...@@ -29,8 +29,8 @@ macro(find_cuda use_cuda)
set(CUDA_FOUND TRUE) set(CUDA_FOUND TRUE)
if(MSVC) if(MSVC)
find_library(CUDA_CUDART_LIBRARY cudart find_library(CUDA_CUDART_LIBRARY cudart
${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib) ${CUDA_TOOLKIT_ROOT_DIR}/lib/Win32)
else(MSVC) else(MSVC)
find_library(CUDA_CUDART_LIBRARY cudart find_library(CUDA_CUDART_LIBRARY cudart
${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib64
...@@ -41,18 +41,18 @@ macro(find_cuda use_cuda) ...@@ -41,18 +41,18 @@ macro(find_cuda use_cuda)
# additional libraries # additional libraries
if(CUDA_FOUND) if(CUDA_FOUND)
if(MSVC) if(MSVC)
find_library(CUDA_NVRTC_LIBRARY cuda find_library(CUDA_CUDA_LIBRARY cuda
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib/win32) ${CUDA_TOOLKIT_ROOT_DIR}/lib/Win32)
find_library(CUDA_NVRTC_LIBRARY nvrtc find_library(CUDA_NVRTC_LIBRARY nvrtc
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib/win32) ${CUDA_TOOLKIT_ROOT_DIR}/lib/Win32)
find_library(CUDA_CUDNN_LIBRARY cudnn find_library(CUDA_CUDNN_LIBRARY cudnn
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib/win32) ${CUDA_TOOLKIT_ROOT_DIR}/lib/Win32)
find_library(CUDA_CUBLAS_LIBRARY cublas find_library(CUDA_CUBLAS_LIBRARY cublas
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib/win32) ${CUDA_TOOLKIT_ROOT_DIR}/lib/Win32)
else(MSVC) else(MSVC)
find_library(_CUDA_CUDA_LIBRARY cuda find_library(_CUDA_CUDA_LIBRARY cuda
PATHS ${CUDA_TOOLKIT_ROOT_DIR} PATHS ${CUDA_TOOLKIT_ROOT_DIR}
......
...@@ -8,9 +8,6 @@ ...@@ -8,9 +8,6 @@
#include <dmlc/thread_local.h> #include <dmlc/thread_local.h>
#include <tvm/runtime/registry.h> #include <tvm/runtime/registry.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <tvm/container.h>
#include <tvm/ir.h>
#include <tvm/packed_func_ext.h>
#include "./cuda_common.h" #include "./cuda_common.h"
namespace tvm { namespace tvm {
......
...@@ -4,9 +4,6 @@ ...@@ -4,9 +4,6 @@
*/ */
#include <tvm/runtime/registry.h> #include <tvm/runtime/registry.h>
#include <dmlc/thread_local.h> #include <dmlc/thread_local.h>
#include <tvm/container.h>
#include <tvm/ir.h>
#include <tvm/packed_func_ext.h>
#include "./sdaccel_common.h" #include "./sdaccel_common.h"
namespace tvm { namespace tvm {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment