Skip to content

Commit f754a28

Browse files
committed
* Fix loading issues with libomp.dylib for PyTorch on macosx-arm64
1 parent 33a7467 commit f754a28

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pytorch/cppbuild.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ ln -sf pytorch/torch/lib ../lib
253253
ln -sf pytorch/torch/bin ../bin
254254

255255
case $PLATFORM in
256-
macosx-*)
256+
macosx-x86_64)
257257
# Disguise libomp as libiomp5 (they share the same codebase and have the same symbols)
258258
# This helps if user wants to link with MKL.
259259
# On linux, user linking with mkl would need to set

pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@
113113
),
114114
@Platform(
115115
value = "macosx",
116-
preload = { "iomp5" }
116+
preload = { "iomp5", "omp" }
117117
),
118118
@Platform(
119119
value = "windows",

0 commit comments

Comments
 (0)