Skip to content

Commit f432594

Browse files
committed
feat: support CUDA GPU Compute Capability 9.0
1 parent b102887 commit f432594

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

core/build.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,7 @@ fn main() {
188188
cc_builder.flag("--expt-relaxed-constexpr");
189189
cc_builder.flag("-gencode=arch=compute_86,code=sm_86");
190190
cc_builder.flag("-gencode=arch=compute_89,code=sm_89");
191+
cc_builder.flag("-gencode=arch=compute_90,code=sm_90");
191192

192193
env::set_var("DEP_SPPARK_ROOT", "../sppark");
193194
if let Some(include) = env::var_os("DEP_SPPARK_ROOT") {

0 commit comments

Comments
 (0)