From fa5bfa1825cb4e74c043a05b60a9224a04070884 Mon Sep 17 00:00:00 2001
From: Diankun An <54262787+piaolaidelangman@users.noreply.github.com>
Date: Tue, 30 Dec 2025 20:49:30 +0800
Subject: [PATCH 1/2] Add CUDA architecture support for compute_100

Fix error "no kernel image is available for execution on the device cuda" in Blackwell GPU
---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b1b08b3..2ca06f9 100644
--- a/setup.py
+++ b/setup.py
@@ -42,6 +42,7 @@
 
 cc_flag = [
     "-gencode", "arch=compute_120a,code=sm_120a", 
+    "-gencode", "arch=compute_100,code=sm_100",
     "-gencode", "arch=compute_90,code=sm_90",
     "-gencode", "arch=compute_89,code=sm_89",
     "-gencode", "arch=compute_80,code=sm_80"
@@ -76,4 +77,4 @@
     ),
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension},
-)
\ No newline at end of file
+)

From 3055aa6d3f1b1bcc3ba554ef63e42ab25d5c29df Mon Sep 17 00:00:00 2001
From: Diankun An <54262787+piaolaidelangman@users.noreply.github.com>
Date: Tue, 30 Dec 2025 21:00:28 +0800
Subject: [PATCH 2/2] Update setup.py