[pytorch] Fixes PyTorch 2.3.1 windows dependencies (#3269)

deepjavalibrary · Jun 20, 2024 · cac16a1 · cac16a1
1 parent 17f992e
commit cac16a1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/engines/pytorch/pytorch-native/build.gradle.kts b/engines/pytorch/pytorch-native/build.gradle.kts
@@ -138,6 +138,7 @@ fun copyNativeLibToOutputDir(fileStoreMap: Map<String, String>, binaryRoot: Stri
                     "cu*.dll",
                     "fbgemm.dll",
                     "nv*.dll",
+                    "mkl_*.dll",
                     "torch_cpu.dll",
                     "torch_cuda*.dll",
                     "torch.dll",