diff options
author | spreis <spreis@yandex-team.com> | 2023-11-16 11:57:47 +0300 |
---|---|---|
committer | spreis <spreis@yandex-team.com> | 2023-11-16 12:32:20 +0300 |
commit | d0bed9523e6cdc3911fcf5ff782e0bcf15d1e393 (patch) | |
tree | 50dcd2d89903d9a64882daf06410cb0ca1ab8668 | |
parent | cad48ec10de00593812e1ff8b2780c203b7d156f (diff) | |
download | ydb-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.conf | 10 |
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 } |