From e5272eec374ea7ff5de15379b5193856b23d5961 Mon Sep 17 00:00:00 2001
From: Consceleratus <135762418+ConsceIeratus@users.noreply.github.com>
Date: Tue, 12 Dec 2023 23:28:28 -0300
Subject: [PATCH] Hopper sm_90a Feature Set Makefile

sm_90a adds support for accelerated wgmma and setmaxnreg instructions.


https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#ptx-module-directives-target
---
 Makefile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index 7ccbcb191..9d8ba61b8 100644
--- a/Makefile
+++ b/Makefile
@@ -51,7 +51,7 @@ CC_cublasLt111 += -gencode arch=compute_80,code=sm_80
 CC_cublasLt111 += -gencode arch=compute_86,code=sm_86
 
 CC_ADA_HOPPER := -gencode arch=compute_89,code=sm_89
-CC_ADA_HOPPER += -gencode arch=compute_90,code=sm_90
+CC_ADA_HOPPER += -gencode arch=compute_90a,code=sm_90a
 
 
 all: $(BUILD_DIR) env