Update VCPKG Triplets for Better ABI Support

Author: root

.clangd

@@ -11,7 +11,6 @@ Diagnostics:
       - cppcoreguidelines-*
       - performance-*
       - google-*
-      - cert-dcl21-cpp
       - cert-dcl58-cpp
       - cert-err58-cpp
       - misc-new-delete-overloads
@@ -53,7 +52,8 @@ Diagnostics:
       readability-identifier-length,
       readability-magic-numbers,
       readability-function-cognitive-complexity,
-      modernize-avoid-c-arrays
+      modernize-avoid-c-arrays,
+      readability-math-missing-parentheses,
     ]
 
     CheckOptions:

.vscode/settings.json

@@ -16,7 +16,7 @@
     "[cpp]": {
         "editor.defaultFormatter": "llvm-vs-code-extensions.vscode-clangd"
     },
-    "clangd.path": "clangd",
+    "clangd.path": "clangd-20",
     "clangd.arguments": [
         "--compile-commands-dir=${workspaceFolder}/build",
         "--log=verbose",

README.md

@@ -9,6 +9,7 @@ Create a new conda environment:
 
 ```bash
 conda create -n cuda-learn python=3.12
+conda upgrade libstdcxx-ng -c conda-forge
 conda activate cuda-learn
 pip install torch torchvision torchaudio
 ```

configs/ctests.yml

@@ -0,0 +1,8 @@
+OpTest:
+  Conv2D:
+    - inputHeight: 32
+      inputWidth: 32
+      kernelSize: 3
+    - inputHeight: 320
+      inputWidth: 320
+      kernelSize: 3

csrc/CMakeLists.txt

@@ -31,6 +31,7 @@ find_package(cxxopts CONFIG REQUIRED)
 find_package(fmt CONFIG REQUIRED)
 find_package(spdlog CONFIG REQUIRED)
 find_package(proxy CONFIG REQUIRED)
+find_package(yaml-cpp CONFIG REQUIRED)
 enable_testing()
 find_package(GTest CONFIG REQUIRED)
 
@@ -46,5 +47,5 @@ add_subdirectory(${PROJECT_SOURCE_DIR}/test)
 
 set(CMAKE_CTEST_COMMAND ctest)
 add_custom_target(check 
-    COMMAND ${CMAKE_CTEST_COMMAND} --output-on-failure --test-dir ${CMAKE_BINARY_DIR}
+    COMMAND ${CMAKE_CTEST_COMMAND} --output-on-failure --test-dir ${CMAKE_CURRENT_BINARY_DIR}
     DEPENDS all)

csrc/cmake/compilers/cuda-compiler-configs.cmake

@@ -22,6 +22,6 @@ set(CMAKE_CUDA_SEPARABLE_COMPILATION ON)
 set(CMAKE_CUDA_ARCHITECTURES native)
 log_info("CMAKE_CUDA_STANDARD: ${CMAKE_CUDA_STANDARD}")
 
-set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-relaxed-constexpr")
-set(CMAKE_CUDA_FLAGS_RELEASE "${CMAKE_CUDA_FLAGS_RELEASE} -O3")
-set(CMAKE_CUDA_FLAGS_DEBUG "${CMAKE_CUDA_FLAGS_DEBUG} -lineinfo")
+string(APPEND CMAKE_CUDA_FLAGS " --expt-relaxed-constexpr")
+string(APPEND CMAKE_CUDA_FLAGS_RELEASE " -O3")
+string(APPEND CMAKE_CUDA_FLAGS_DEBUG " -lineinfo")

csrc/cmake/libraries/libcuda.cmake

@@ -11,11 +11,11 @@ include(${CMAKE_CURRENT_LIST_DIR}/../utils/common.cmake)
 
 try_get_value(CUDA_HOME HINTS CUDA_HOME CUDA_DIR)
 if (NOT CUDA_HOME_FOUND)
-    log_error("ENV:CUDA_HOME is not set.")
+    log_error("CUDA_HOME not set.")
 endif()
 
 # Append the path to the CMAKE_PREFIX_PATH
-set(CUDA_CMAKE_PREFIX_PATH "$CUDA_HOME/lib64/cmake")
+set(CUDA_CMAKE_PREFIX_PATH "${CUDA_HOME}/lib64/cmake")
 list(APPEND CMAKE_PREFIX_PATH ${CUDA_CMAKE_PREFIX_PATH})
 
 # Find the CUDA package

csrc/cmake/utils/run-python.cmake

@@ -1,5 +1,5 @@
-include(${PROJECT_SOURCE_DIR}/cmake/utils/logging.cmake)
-include(${PROJECT_SOURCE_DIR}/cmake/utils/common.cmake)
+include(${CMAKE_CURRENT_LIST_DIR}/logging.cmake)
+include(${CMAKE_CURRENT_LIST_DIR}/common.cmake)
 
 
 # @brief Run python command and get result, output and error

csrc/cmake/vcpkg-triplets/x64-linux-no-cxx11abi.cmake

@@ -0,0 +1,23 @@
+set(VCPKG_TARGET_ARCHITECTURE x64)
+set(VCPKG_CRT_LINKAGE dynamic)
+set(VCPKG_LIBRARY_LINKAGE static)
+
+set(VCPKG_CMAKE_SYSTEM_NAME Linux)
+
+execute_process(
+    COMMAND python -c "import torch; print(torch.__version__.split('+')[0])"
+    OUTPUT_VARIABLE TORCH_VERSION
+    OUTPUT_STRIP_TRAILING_WHITESPACE
+)
+
+# Convert version string to comparable number
+string(REPLACE "." ";" VERSION_LIST ${TORCH_VERSION})
+list(GET VERSION_LIST 0 TORCH_MAJOR)
+list(GET VERSION_LIST 1 TORCH_MINOR)
+math(EXPR TORCH_VERSION_NUM "${TORCH_MAJOR} * 1000 + ${TORCH_MINOR}")
+
+# Set ABI flag only if torch version < 2.6
+if(TORCH_VERSION_NUM LESS 2006)
+    set(ENV{CXXFLAGS} "$ENV{CXXFLAGS} -D_GLIBCXX_USE_CXX11_ABI=0")
+    set(ENV{CFLAGS} "$ENV{CFLAGS} -D_GLIBCXX_USE_CXX11_ABI=0")
+endif()