diff options
| author | deshevoy <deshevoy@yandex-team.com> | 2024-12-11 01:13:42 +0300 |
|---|---|---|
| committer | deshevoy <deshevoy@yandex-team.com> | 2024-12-11 01:28:20 +0300 |
| commit | 7883b5b26679a55579418c4df15961961c8b9cbc (patch) | |
| tree | 3dbbcb5e7e3a8d4fb7de7765c992a1e98ea3cf44 | |
| parent | c415f030b477653db847de5dbc58cb579267c1da (diff) | |
| download | ydb-7883b5b26679a55579418c4df15961961c8b9cbc.tar.gz | |
[build] cuda: Fix dynamic linking with new cuDNN versions
commit_hash:0472b64d3ba1e788aaee0ca8ab2fd21c9d2262dd
| -rw-r--r-- | build/scripts/link_exe.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/build/scripts/link_exe.py b/build/scripts/link_exe.py index 634840cf90..12f2cb1083 100644 --- a/build/scripts/link_exe.py +++ b/build/scripts/link_exe.py @@ -34,6 +34,12 @@ CUDA_LIBRARIES = { '-lcublasLt_static': '-lcublasLt', '-lcudart_static': '-lcudart', '-lcudnn_static': '-lcudnn', + '-lcudnn_adv_infer_static': '-lcudnn', + '-lcudnn_adv_train_static': '-lcudnn', + '-lcudnn_cnn_infer_static': '-lcudnn', + '-lcudnn_cnn_train_static': '-lcudnn', + '-lcudnn_ops_infer_static': '-lcudnn', + '-lcudnn_ops_train_static': '-lcudnn', '-lcufft_static_nocallback': '-lcufft', '-lcupti_static': '-lcupti', '-lcurand_static': '-lcurand', |