Skip to content

Commit 74b6b86

Browse files
author
Niklas Gustafsson
committed
Updated list of preloaded CUDNN DLLs.
1 parent 4676ec4 commit 74b6b86

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/TorchSharp/Torch.cs

+3
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,9 @@ private static void LoadNativeBackend(bool useCudaBackend, out StringBuilder? tr
114114
ok = TryLoadNativeLibraryByName("cudnn_adv64_9", typeof(torch).Assembly, trace);
115115
ok = TryLoadNativeLibraryByName("cudnn_cnn64_9", typeof(torch).Assembly, trace);
116116
ok = TryLoadNativeLibraryByName("cudnn_ops64_9", typeof(torch).Assembly, trace);
117+
ok = TryLoadNativeLibraryByName("cudnn_graph64_9.dll", typeof(torch).Assembly, trace);
118+
ok = TryLoadNativeLibraryByName("cudnn_engines_precompiled64_9.dll", typeof(torch).Assembly, trace);
119+
ok = TryLoadNativeLibraryByName("cudnn_engines_runtime_compiled64_9.dll", typeof(torch).Assembly, trace);
117120
ok = TryLoadNativeLibraryByName("nvrtc-builtins64_121", typeof(torch).Assembly, trace);
118121
ok = TryLoadNativeLibraryByName("caffe2_nvrtc", typeof(torch).Assembly, trace);
119122
ok = TryLoadNativeLibraryByName("nvrtc64_120_0", typeof(torch).Assembly, trace);

0 commit comments

Comments
 (0)