From 3a6b475cef395591958dd27bc4e90c0a005c22aa Mon Sep 17 00:00:00 2001
From: Shubham <32878682+IamShubhamGupto@users.noreply.github.com>
Date: Fri, 22 Mar 2024 23:43:30 -0400
Subject: [PATCH] nvidia jetson hw support

https://developer.nvidia.com/cuda-gpus
---
 setup.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/setup.py b/setup.py
index 7d89db4..142addc 100644
--- 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")