Skip to content

Commit

Permalink
Link to libomp full name (#558)
Browse files Browse the repository at this point in the history
  • Loading branch information
karllessard authored Sep 20, 2024
1 parent 0440fc6 commit 8aa4861
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ if [[ "$PLATFORM" =~ "linux" ]]; then
ln -fs libtensorflow_cc.so.2 libtensorflow_cc.so
ln -fs libtensorflow_framework.so.2 libtensorflow_framework.so
if [[ "$PLATFORM" == "linux-arm64" ]]; then
ln -fs libomp-*.so.5 libomp.so
ln -fs libomp-e9212f90.so.5 libomp-e9212f90.so
fi
elif [[ "$PLATFORM" =~ "macosx" ]]; then
ln -fs libtensorflow_cc.2.dylib libtensorflow_cc.dylib
Expand All @@ -65,4 +65,4 @@ elif [[ "$PLATFORM" =~ "windows" ]]; then
# (while it is also available at the root of the include folder for other platforms)
cd include && ln -fs tensorflow/tsl tsl && cd -
fi
ls -l .
ls -l .
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@
resource = {"LICENSE", "THIRD_PARTY_TF_JNI_LICENSES"}),
@Platform(
value = {"linux-arm64"},
link = {"tensorflow_cc@.2", "tensorflow_framework@.2", "omp@.5"}),
link = {"tensorflow_cc@.2", "tensorflow_framework@.2", "omp-e9212f90@.5"}),
@Platform(
value = "windows",
preload = {
Expand Down

0 comments on commit 8aa4861

Please sign in to comment.