From b6a19ad8812700c66d38ccbb12aba8d4f72692e2 Mon Sep 17 00:00:00 2001 From: Freax Ruby Date: Tue, 19 Aug 2025 02:37:58 -0400 Subject: [PATCH] Update setup.py --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 275b6fc3..3e14ad59 100644 --- a/setup.py +++ b/setup.py @@ -92,6 +92,7 @@ def get_extensions(): "-D__CUDA_NO_HALF_OPERATORS__", "-D__CUDA_NO_HALF_CONVERSIONS__", "-D__CUDA_NO_HALF2_OPERATORS__", + "-gencode=arch=compute_120,code=sm_120", ] else: print("Compiling without CUDA")