Refactor static loop.

f625455c · Andriy Roshchenko · 128685c3 · f625455c
Commit f625455c authored Jan 30, 2025 by Andriy Roshchenko
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

test/mx_mfma_op/mx_mfma_op.hpp test/mx_mfma_op/mx_mfma_op.hpp +6 -4

No files found.
--- a/test/mx_mfma_op/mx_mfma_op.hpp
+++ b/test/mx_mfma_op/mx_mfma_op.hpp
@@ -129,10 +129,12 @@ __device__ AFragT load_A_col_major(AType const* input_ptr)
        bit_cast<ARawT>(input_ptr[startOffset + 31 * kOffset])}; // XXX v[31] = Reg 7 [24:31]
 #else
    auto fragA = AScalarFragT{};
-    static_for<0, VW, 1>{}([&](auto i) {
+#pragma unroll VW
-        fragA[static_cast<int>(i)] =
+    for(uint32_t i = 0; i < VW; i++)
-            bit_cast<ARawT>(input_ptr[startOffset + static_cast<int>(i) * kOffset]);
+    {
-    });
+        fragA[i] = bit_cast<ARawT>(input_ptr[startOffset + i * kOffset]);
+    }
 #endif
    return fragA;
 }