From 6b1b4516893f880fb8278093c8aeeae8c663c128 Mon Sep 17 00:00:00 2001 From: Daniel Vega-Myhre Date: Wed, 10 Sep 2025 19:11:39 -0700 Subject: [PATCH 1/2] Add nvcc flags for building MXFP8 dim1 cast kernel for sm100a on CPU-only build runners --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index c6b60030a5..775a601ed5 100644 --- 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"], }, ), ) From 8514b11ddac94d2fa983b41f4e988ed9c283e9d3 Mon Sep 17 00:00:00 2001 From: Daniel Vega-Myhre Date: Thu, 11 Sep 2025 09:55:37 -0700 Subject: [PATCH 2/2] build for sm100 and sm120 --- setup.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 775a601ed5..fd4ee9f40f 100644 --- a/setup.py +++ b/setup.py @@ -677,7 +677,11 @@ def get_extensions(): ], extra_compile_args={ "cxx": ["-std=c++17", "-O3"], - "nvcc": nvcc_args + ["-gencode=arch=compute_100a,code=sm_100a"], + "nvcc": nvcc_args + + [ + "-gencode=arch=compute_100,code=sm_100", + "-gencode=arch=compute_120,code=compute_120", + ], }, ), )