fix(gpu): fix tfhe-cuda-backend release

zama-ai · Mar 26, 2024 · 2955f0a · 2955f0a
1 parent f5fb578
commit 2955f0a
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 7 deletions.
diff --git a/backends/tfhe-cuda-backend/cuda/CMakeLists.txt b/backends/tfhe-cuda-backend/cuda/CMakeLists.txt
@@ -65,9 +65,7 @@ else()
   set(CUDA_ARCH "700")
 endif()
 
-message(STATUS "CUDA_ARCH: ${CUDA_ARCH}")
-
-configure_file(${CMAKE_CURRENT_SOURCE_DIR}/cuda_config.h.in ${CMAKE_CURRENT_SOURCE_DIR}/include/cuda_config.h)
+add_compile_definitions(CUDA_ARCH=${CUDA_ARCH})
 
 # in production, should use -arch=sm_70 --ptxas-options=-v to see register spills -lineinfo for better debugging
 set(CMAKE_CUDA_FLAGS

diff --git a/backends/tfhe-cuda-backend/cuda/cuda_config.h.in b/backends/tfhe-cuda-backend/cuda/cuda_config.h.in
diff --git a/backends/tfhe-cuda-backend/cuda/include/device.h b/backends/tfhe-cuda-backend/cuda/include/device.h
@@ -1,7 +1,6 @@
 #ifndef DEVICE_H
 #define DEVICE_H
 
-#include "cuda_config.h"
 #include <cstdint>
 #include <cstdio>
 #include <cstdlib>