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 17, 2024
1 parent cfc82c1 commit 18c6ec8
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 2 deletions.
1 change: 1 addition & 0 deletions pytorch/platform/gpu/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,7 @@
</dependency>
</dependencies>

<!-- To help finding parent pom, but probably useless -->
<repositories>
<repository>
<id>sonatype-nexus-snapshots</id>
Expand Down
1 change: 1 addition & 0 deletions pytorch/platform/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@
</dependency>
</dependencies>

<!-- To help finding parent pom, but probably useless -->
<repositories>
<repository>
<id>sonatype-nexus-snapshots</id>
Expand Down
1 change: 1 addition & 0 deletions pytorch/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@
</dependency>
</dependencies>

<!-- To help finding parent pom, but probably useless -->
<repositories>
<repository>
<id>sonatype-nexus-snapshots</id>
Expand Down
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 18c6ec8

Please sign in to comment.