Skip to content

Commit c1055d8

Browse files
authored
Have CUDA compilaion check for sbsa-linux when targeting aarch64. (#1647)
Originally, CUDA placed aarch64 libraries in a "natural" path, {target_arch}-linux, but then changed it to sbsa-linux.
1 parent 7abc2e6 commit c1055d8

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/lib.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1672,6 +1672,10 @@ impl Build {
16721672
if cfg!(target_os = "linux") {
16731673
libdir.push("targets");
16741674
libdir.push(format!("{}-linux", target.arch));
1675+
if !libdir.exists() && target.arch == "aarch64" {
1676+
libdir.pop();
1677+
libdir.push("sbsa-linux");
1678+
}
16751679
libdir.push("lib");
16761680
libtst = true;
16771681
} else if cfg!(target_env = "msvc") {

0 commit comments

Comments
 (0)