squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add missing checks for NVRTC in CuTe (#921)

Browse Source
This commit is contained in:
Jakub Szuppe 2023-04-25 18:52:43 +02:00 committed by GitHub
parent e36912f961
commit 180c5629bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/cute/arch/copy_sm90_desc.hpp
View File

@ -134,6 +134,7 @@ enum class SmemSwizzleBits : uint8_t {
B128 = 3, B128 = 3,
}; };
#if !defined(__CUDACC_RTC__)
#if (__CUDACC_VER_MAJOR__ >= 12) #if (__CUDACC_VER_MAJOR__ >= 12)
template <class T> template <class T>
@ -164,9 +165,10 @@ inline CUtensorMapSwizzle to_CUtensorMapSwizzle(SmemSwizzleBits const& t) {
} }
#endif // (__CUDACC_VER_MAJOR__ >= 12) #endif // (__CUDACC_VER_MAJOR__ >= 12)
#endif // !defined(__CUDACC_RTC__)
} // end namespace TMA } // end namespace TMA
#if (__CUDACC_VER_MAJOR__ >= 12) #if (__CUDACC_VER_MAJOR__ >= 12) && !defined(__CUDACC_RTC__)
using TmaDescriptor = CUtensorMap; using TmaDescriptor = CUtensorMap;
#else #else
using TmaDescriptor = struct { char bytes[128]; }; using TmaDescriptor = struct { char bytes[128]; };