Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
bitsandbytes
Commits
e4a9a94c
Commit
e4a9a94c
authored
Feb 24, 2025
by
Matthew Douglas
Browse files
Update build-cuda.sh
parent
12cc3fb4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
7 deletions
+10
-7
.github/scripts/build-cuda.sh
.github/scripts/build-cuda.sh
+10
-7
No files found.
.github/scripts/build-cuda.sh
View file @
e4a9a94c
...
@@ -4,13 +4,16 @@ declare build_os
...
@@ -4,13 +4,16 @@ declare build_os
declare
cuda_version
declare
cuda_version
set
-xeuo
pipefail
set
-xeuo
pipefail
build_capability
=
"50;52;60;61;70;75;80;86;89;90;100;120"
remove_for_11_7
=
";89;90;100;120"
# By default, target Maxwell through Hopper.
remove_for_11_8
=
";100;120"
build_capability
=
"50;52;60;61;70;75;80;86;89;90"
remove_for_lt_12_7
=
";100;120"
[[
"
${
cuda_version
}
"
==
11.7.
*
]]
&&
build_capability
=
$(
sed
's|'
"
$remove_for_11_7
"
'||g'
<<<
"
$build_capability
"
)
# CUDA 11.7: Remove sm89 and sm90
[[
"
${
cuda_version
}
"
==
11.8.
*
]]
&&
build_capability
=
$(
sed
's|'
"
$remove_for_11_8
"
'||g'
<<<
"
$build_capability
"
)
[[
"
${
cuda_version
}
"
==
11.7.
*
]]
&&
build_capability
=
"50;52;60;61;70;75;80;86"
[[
"
${
cuda_version
}
"
< 12.7
]]
&&
build_capability
=
$(
sed
's|'
"
$remove_for_lt_12_7
"
'||g; s|'
"
${
remove_for_lt_12_7
#;
}
"
';||g'
<<<
"
$build_capability
"
)
# CUDA 12.8: Add sm100 and sm120; remove sm50 through sm61
[[
"
${
cuda_version
}
"
==
12.8.
*
]]
&&
build_capability
=
"70;75;80;86;89;90;100;120"
[[
"
${
build_os
}
"
=
windows-
*
]]
&&
python3
-m
pip
install
ninja
[[
"
${
build_os
}
"
=
windows-
*
]]
&&
python3
-m
pip
install
ninja
if
[
"
${
build_os
:0:6
}
"
==
ubuntu
]
;
then
if
[
"
${
build_os
:0:6
}
"
==
ubuntu
]
;
then
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment