P6: Fix host-side BF16→FP32 conversion in test

2026-05-30 17:01:13 +00:00
parent a88b321433
commit e4ee9fdc9f
1 changed files with 7 additions and 1 deletions
--- a/tests/unit/test_p6_tma_store.cu
+++ b/tests/unit/test_p6_tma_store.cu
@@ -153,7 +153,13 @@ int main() {

    float* f_direct = new float[n_h * HD];
    float* f_tma = new float[n_h * HD];
-        auto b2f = [](bf16_t h) -> float { float f; asm("cvt.f32.bf16 %0, %1;" : "=f"(f) : "h"(h)); return f; };
+        auto b2f = [](bf16_t h) -> float {
+            unsigned short us = h;
+            unsigned int u = us << 16;
+            float f;
+            memcpy(&f, &u, 4);
+            return f;
+        };
        for (int i = 0; i < n_h * HD; i++) {
            f_direct[i] = b2f(h_o_direct[i]);
            f_tma[i] = b2f(h_o_tma[i]);