Merge pull request #20 from IamShubhamGupto/main

nvidia jetson hw support
Dao-AILab · Mar 23, 2024 · 3db8bef · 3db8bef
2 parents 9645672 + 3a6b475
commit 3db8bef
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -101,10 +101,18 @@ def append_nvcc_threads(nvcc_extra_args):
                 "Note: make sure nvcc has a supported version by running nvcc -V."
             )
 
+    cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_53,code=sm_53")
+    cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_62,code=sm_62")
     cc_flag.append("-gencode")
     cc_flag.append("arch=compute_70,code=sm_70")
     cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_72,code=sm_72")
+    cc_flag.append("-gencode")
     cc_flag.append("arch=compute_80,code=sm_80")
+    cc_flag.append("-gencode")
+    cc_flag.append("arch=compute_87,code=sm_87")
     if bare_metal_version >= Version("11.8"):
         cc_flag.append("-gencode")
         cc_flag.append("arch=compute_90,code=sm_90")