Skip to content

Commit

Permalink
Add uv to preloads
Browse files Browse the repository at this point in the history
  • Loading branch information
HGuillemet committed Jun 20, 2024
1 parent 3ba3ac2 commit 7abb739
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java
Original file line number Diff line number Diff line change
Expand Up @@ -86,12 +86,12 @@
},
exclude = {"openblas_config.h", "cblas.h", "lapacke_config.h", "lapacke_mangling.h", "lapack.h", "lapacke.h", "lapacke_utils.h"},
link = {"c10", "torch_cpu", "torch"},
preload = {"[email protected]", "iomp5", "omp", "[email protected]", "asmjit", "fbgemm"}
preload = {"[email protected]", "iomp5", "omp", "[email protected]", "asmjit", "fbgemm", "uv"}
),
@Platform(
value = {"linux", "macosx", "windows"},
link = { "c10", "c10_cuda", "torch_cpu", "torch_cuda", "torch" },
preload = {"[email protected]", "iomp5", "omp", "[email protected]", "asmjit", "fbgemm", "[email protected]"},
preload = {"[email protected]", "iomp5", "omp", "[email protected]", "asmjit", "fbgemm", "uv", "[email protected]"},
includepath = {"/usr/local/cuda/include", "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.3/include/"},
preloadpath = {
"/usr/local/cuda-12.3/lib64/",
Expand Down

0 comments on commit 7abb739

Please sign in to comment.