Skip to content

Commit f1e118b

Browse files
Add nvcc flags to explicitly build mxfp8 dim1 cast kernel for sm100a (#2979)
* Add nvcc flags for building MXFP8 dim1 cast kernel for sm100a on CPU-only build runners * build for sm100 and sm120
1 parent be71434 commit f1e118b

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

setup.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -677,7 +677,11 @@ def get_extensions():
677677
],
678678
extra_compile_args={
679679
"cxx": ["-std=c++17", "-O3"],
680-
"nvcc": nvcc_args,
680+
"nvcc": nvcc_args
681+
+ [
682+
"-gencode=arch=compute_100,code=sm_100",
683+
"-gencode=arch=compute_120,code=compute_120",
684+
],
681685
},
682686
),
683687
)

0 commit comments

Comments
 (0)