Skip to content

Commit 294ff33

Browse files
author
Niklas Gustafsson
committed
More tinkering.
1 parent 74b6b86 commit 294ff33

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/TorchSharp/Torch.cs

+1
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ private static void LoadNativeBackend(bool useCudaBackend, out StringBuilder? tr
115115
ok = TryLoadNativeLibraryByName("cudnn_cnn64_9", typeof(torch).Assembly, trace);
116116
ok = TryLoadNativeLibraryByName("cudnn_ops64_9", typeof(torch).Assembly, trace);
117117
ok = TryLoadNativeLibraryByName("cudnn_graph64_9.dll", typeof(torch).Assembly, trace);
118+
ok = TryLoadNativeLibraryByName("cudnn_heuristic64_9.dll", typeof(torch).Assembly, trace);
118119
ok = TryLoadNativeLibraryByName("cudnn_engines_precompiled64_9.dll", typeof(torch).Assembly, trace);
119120
ok = TryLoadNativeLibraryByName("cudnn_engines_runtime_compiled64_9.dll", typeof(torch).Assembly, trace);
120121
ok = TryLoadNativeLibraryByName("nvrtc-builtins64_121", typeof(torch).Assembly, trace);

0 commit comments

Comments
 (0)