Add nvcc flags for building MXFP8 dim1 cast kernel for sm100a on CPU-only build runners

danielvegamyhre · danielvegamyhre · commit 6b1b4516893f · 2025-09-10T19:11:39.000-07:00
diff --git a/setup.py b/setup.py
@@ -677,7 +677,7 @@ def get_extensions():
                     ],
                     extra_compile_args={
                         "cxx": ["-std=c++17", "-O3"],
-                        "nvcc": nvcc_args,
+                        "nvcc": nvcc_args + ["-gencode=arch=compute_100a,code=sm_100a"],
                     },
                 ),
             )

Original file line number	Diff line number	Diff line change
`@@ -677,7 +677,7 @@ def get_extensions():`
`677`	`677`	`],`
`678`	`678`	`extra_compile_args={`
`679`	`679`	`"cxx": ["-std=c++17", "-O3"],`
`680`		`- "nvcc": nvcc_args,`
	`680`	`+ "nvcc": nvcc_args + ["-gencode=arch=compute_100a,code=sm_100a"],`
`681`	`681`	`},`
`682`	`682`	`),`
`683`	`683`	`)`