Skip to content

Commit

Permalink
Split JNI libraries
Browse files Browse the repository at this point in the history
  • Loading branch information
HGuillemet committed Aug 9, 2024
1 parent cef1268 commit bdf9805
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 10 deletions.
8 changes: 0 additions & 8 deletions pytorch/src/main/java/org/bytedeco/pytorch/presets/gloo.java
Original file line number Diff line number Diff line change
Expand Up @@ -23,23 +23,15 @@

import org.bytedeco.javacpp.ClassProperties;
import org.bytedeco.javacpp.LoadEnabled;
import org.bytedeco.javacpp.annotation.Platform;
import org.bytedeco.javacpp.annotation.Properties;
import org.bytedeco.javacpp.presets.chrono;
import org.bytedeco.javacpp.tools.*;

import static org.bytedeco.pytorch.presets.torch.template;

/**
* @author Hervé Guillemet
*/
@Properties(
inherit = { torch.class, chrono.class },
value = {
@Platform(
library = "jnitorch"
)
},
target = "org.bytedeco.pytorch.gloo",
global = "org.bytedeco.pytorch.global.gloo"
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,7 @@
exclude = {
"<cublas.h>", // pytorch includes cublas_v2, which is not compatible with cublas included from inherited cudnn presets
"<driver_functions.h>" // causes #warning
},
library = "jnitorch"
}
),
},
target = "org.bytedeco.pytorch.cuda",
Expand Down

0 comments on commit bdf9805

Please sign in to comment.