Skip to content

Commit

Permalink
[pytorch] Ensure shared library loading order for aarch64
Browse files Browse the repository at this point in the history
  • Loading branch information
frankfliu committed Dec 12, 2023
1 parent 3c1301a commit 869afe6
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,8 @@ private static void loadLibTorch(LibTorch libTorch) {
&& name.contains("cudart")
&& name.contains("nvTools")) {
return false;
} else if (name.startsWith("libarm_compute-")) {
} else if (name.startsWith("libarm_compute-")
|| name.startsWith("libopenblasp")) {
rank.put(path, 2);
return true;
} else if (name.startsWith("libarm_compute_")) {
Expand Down

0 comments on commit 869afe6

Please sign in to comment.