From 180c5629bfa3889cf7afb259b362d19eb5f24cbe Mon Sep 17 00:00:00 2001
From: Jakub Szuppe <j.szuppe@gmail.com>
Date: Tue, 25 Apr 2023 18:52:43 +0200
Subject: [PATCH] Add missing checks for NVRTC in CuTe (#921)

---
 include/cute/arch/copy_sm90_desc.hpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/include/cute/arch/copy_sm90_desc.hpp b/include/cute/arch/copy_sm90_desc.hpp
index f69c2bd5..489998b5 100644
--- a/include/cute/arch/copy_sm90_desc.hpp
+++ b/include/cute/arch/copy_sm90_desc.hpp
@@ -134,6 +134,7 @@ enum class SmemSwizzleBits : uint8_t {
   B128 = 3,
 };
 
+#if !defined(__CUDACC_RTC__)
 #if (__CUDACC_VER_MAJOR__ >= 12)
 
 template <class T>
@@ -164,9 +165,10 @@ inline CUtensorMapSwizzle to_CUtensorMapSwizzle(SmemSwizzleBits const& t) {
 }
 
 #endif // (__CUDACC_VER_MAJOR__ >= 12)
+#endif // !defined(__CUDACC_RTC__)
 } // end namespace TMA
 
-#if (__CUDACC_VER_MAJOR__ >= 12)
+#if (__CUDACC_VER_MAJOR__ >= 12) && !defined(__CUDACC_RTC__)
 using TmaDescriptor = CUtensorMap;
 #else
 using TmaDescriptor = struct { char bytes[128]; };