Skip to content

Commit 404fc00

Browse files
committed
feat: support CUDA GPU Compute Capability 9.0
1 parent ef5b821 commit 404fc00

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
@@ -189,6 +189,7 @@ fn main() {
189189
cc_builder.flag("-gencode=arch=compute_86,code=sm_86");
190190
cc_builder.flag("-gencode=arch=compute_89,code=sm_89");
191191
cc_builder.flag("-gencode=arch=compute_89,code=compute_89");
192+
cc_builder.flag("-gencode=arch=compute_90,code=sm_90");
192193

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

0 commit comments

Comments
 (0)