Define TensorMap for CUDA 11 (#932)

NVIDIA · Sep 23, 2023 · 8facc54 · 8facc54
1 parent 493913e
commit 8facc54
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/csrc/tma.cpp b/csrc/tma.cpp
@@ -104,6 +104,13 @@ inline CUtensorMapFloatOOBfill getCUtensorMapFloatOOBfill(
 
 using TensorMap = CUtensorMap;
 
+#else
+
+// Placeholder for CUDA 11 to make build pass
+struct TensorMap {
+  alignas(64) uint64_t opaque[16];
+};
+
 #endif
 
 std::ostream& operator<<(std::ostream& os, TensorMapInterleave interleave) {