aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorspreis <spreis@yandex-team.com>2023-11-16 11:57:47 +0300
committerspreis <spreis@yandex-team.com>2023-11-16 12:32:20 +0300
commitd0bed9523e6cdc3911fcf5ff782e0bcf15d1e393 (patch)
tree50dcd2d89903d9a64882daf06410cb0ca1ab8668
parentcad48ec10de00593812e1ff8b2780c203b7d156f (diff)
downloadydb-d0bed9523e6cdc3911fcf5ff782e0bcf15d1e393.tar.gz
Add CUDA11 and CUDA12 presets and fix CUDNN 8.6.0 static build
- This will allow to have multiple CUDA defaults and so simplify migration - CuDNN was split into multiple static libraries. Let's link them all by now (we may inroduce fine-grain control later) - CUDA12.1 with default OS_SDK fails to link even with lowest possible CuDNN and TensorRT: ``` [=] ya make dict/mt/libs/nn/ynmt/gtest/gpu -DCUDA_VERSION=12.1 -DCUDNN_VERSION=8.0.5 -DTENSORRT_VERSION=7 . . . ld.lld: error: undefined symbol: std::invalid_argument::invalid_argument(char const*) >>> referenced by builtin.cpp >>> _0_4_builtin.cpp.o:(cask__5x_cublas::ir::CreateOrGetTensorArgument(cask__5x_cublas::ir::Namespace&, cask__5x_cublas::ir::NumericType*, cask__5x_cublas::ir::LayoutType*)) in archive /home/spreis/.ya/tools/v4/5048904205/lib64/libcublasLt_static.a >>> did you mean: std::invalid_argument::invalid_argument(char const*) >>> defined in: dict/dictutil/libdict-dictutil.a(exceptions.cpp.o) clang-14: error: linker command failed with exit code 1 (use -v to see invocation) Failed [=] ya make dict/mt/libs/nn/ynmt/gtest/gpu -DCUDA_VERSION=12.1 -DCUDNN_VERSION=8.0.5 -DTENSORRT_VERSION=7 -DOS_SDK=ubuntu-20 Ok ```
-rw-r--r--build/ymake.core.conf10
1 files changed, 10 insertions, 0 deletions
diff --git a/build/ymake.core.conf b/build/ymake.core.conf
index eb62bcf885..7cf3a9536f 100644
--- a/build/ymake.core.conf
+++ b/build/ymake.core.conf
@@ -5435,11 +5435,21 @@ TEST_TOOL_HOST_LOCAL=
TEST_TOOL_TARGET_LOCAL=
when ($MOST_USED_CUDA) {
+ CUDA11=yes
+}
+
+when ($CUDA11) {
CUDA_VERSION=11.4
CUDNN_VERSION=8.0.5
TENSORRT_VERSION=7
}
+when ($CUDA12) {
+ CUDA_VERSION=12.1
+ CUDNN_VERSION=8.6.0
+ TENSORRT_VERSION=8
+}
+
when ($TENSORFLOW_WITH_CUDA) {
CUDA_REQUIRED=yes
}