remove cudnn

microsoft · Apr 19, 2023 · e4e02fe · e4e02fe
1 parent 246385a
commit e4e02fe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git a/src/nnfusion/core/kernels/cuda_gpu/cuda_langunit.cpp b/src/nnfusion/core/kernels/cuda_gpu/cuda_langunit.cpp
@@ -10,7 +10,7 @@ using namespace nnfusion::kernels;
 // Header
 LU_DEFINE(header::cuda, "#include <cuda.h>\n#include <cuda_runtime.h>\n#include <cooperative_groups.h>\n");
 LU_DEFINE(header::cublas, "#include <cublas_v2.h>\n");
-LU_DEFINE(header::cudnn, "#include <cudnn.h>\n");
+LU_DEFINE(header::cudnn, "// #include <cudnn.h>\n");
 LU_DEFINE(header::superscaler, "#include \"superscaler.h\"\n");
 LU_DEFINE(header::cupti, "#include <cupti.h>\n");
 LU_DEFINE(header::cuda_prof_api, "#include <cuda_profiler_api.h>\n");

diff --git a/src/nnfusion/engine/pass/codegen/codegen_langunit.cpp b/src/nnfusion/engine/pass/codegen/codegen_langunit.cpp
@@ -63,15 +63,7 @@ LU_DEFINE(nnfusion::codegen::cmake::cuda_lib,
           R"(
 link_directories(${CUDA_TOOLKIT_ROOT_DIR}/lib64)
 
-find_path(CUDNN_INCLUDE_DIR cudnn.h
-    HINTS ${CUDA_TOOLKIT_ROOT_DIR}
-    PATH_SUFFIXES cuda/include include)
-
-include_directories(${CUDNN_INCLUDE_DIR} ${CUDA_INCLUDE_DIRS})
-
-find_library(CUDNN_LIBRARY cudnn
-    HINTS ${CUDA_TOOLKIT_ROOT_DIR}
-    PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64)
+include_directories(${CUDA_INCLUDE_DIRS})
 
 find_library(CUDA_cuda_LIBRARY cuda ${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs)
 find_library(CUDA_cudart_LIBRARY libcudart.so ${CUDA_TOOLKIT_ROOT_DIR}/lib64)
@@ -80,8 +72,7 @@ target_link_libraries(${TARGET_NAME}
     ${CUDA_cuda_LIBRARY}
     ${CUDA_cudart_LIBRARY}
     ${CUDA_LIBRARIES}
-    ${CUDA_CUBLAS_LIBRARIES}
-    ${CUDNN_LIBRARY})
+    ${CUDA_CUBLAS_LIBRARIES})
 )");
 
 LU_DEFINE(nnfusion::codegen::cmake::rocm_lib,