From fe10e1c5d97bbc97e970060c35961927403775bf Mon Sep 17 00:00:00 2001
From: Ajay Panyala <ajay.panyala@gmail.com>
Date: Thu, 30 May 2024 21:24:40 -0700
Subject: [PATCH] cutlass requires cuda CC >= 8.0

---
 cmake/gauxc-cutlass.cmake | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cmake/gauxc-cutlass.cmake b/cmake/gauxc-cutlass.cmake
index 5f5bf571..7020eb2f 100644
--- a/cmake/gauxc-cutlass.cmake
+++ b/cmake/gauxc-cutlass.cmake
@@ -1,7 +1,7 @@
 # Check that only CUDA CC 8.0+ is enabled
 foreach( cuda_arch ${CMAKE_CUDA_ARCHITECTURES} )
-  if( NOT cuda_arch EQUAL 80 )
-    message(FATAL_ERROR "GauXC Requires CUDA CC 8.0 For CUTLASS")
+  if( NOT cuda_arch GREATER_EQUAL 80 )
+    message(FATAL_ERROR "GauXC Requires CUDA CC >= 8.0 For CUTLASS")
   endif()
 endforeach()