[llvm] [NVPTX] Add im2colw/w128 modes support to TMA intrinsics (PR #148863)

Artem Belevich via llvm-commits llvm-commits at lists.llvm.org
Thu Jul 17 11:07:19 PDT 2025


================
@@ -782,6 +922,15 @@ foreach dim = 1...5 in {
     defm TMA_TENSOR_PF_ # suffix : TMA_TENSOR_PREFETCH_INTR<dim, mode>;
   }
 }
+foreach dim = 3...5 in {
+  foreach mode = ["im2col_w", "im2col_w_128"] in {
+    defvar suffix = !toupper(mode) # "_" # dim # D;
----------------
Artem-B wrote:

Nit: here and in other places make it  `"D"`, so it's clear that it's a string literal, similar to `"_"`.

https://github.com/llvm/llvm-project/pull/148863


More information about the llvm-commits mailing list