Skip to content

Commit

Permalink
Use full libomp name
Browse files Browse the repository at this point in the history
  • Loading branch information
karllessard authored Sep 20, 2024
1 parent f211301 commit cf17c2e
Showing 1 changed file with 2 additions and 2 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.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 .

0 comments on commit cf17c2e

Please sign in to comment.