Fix TMA: use CU_TENSOR_MAP_DATA_TYPE_BFLOAT16 not UINT16

2026-05-28 16:51:40 +00:00
parent 0fd3e12a52
commit 55d289c65b
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_tma_driver.cu
+++ b/tests/unit/test_tma_driver.cu
@@ -92,7 +92,7 @@ int main() {

    CUtensorMap tma_desc_host;
    CUresult res = cuTensorMapEncodeTiled(
-        &tma_desc_host, CU_TENSOR_MAP_DATA_TYPE_UINT16, 2,
+        &tma_desc_host, CU_TENSOR_MAP_DATA_TYPE_BFLOAT16, 2,
        d_src, gdim, gstr, tdim, tstr,
        CU_TENSOR_MAP_INTERLEAVE_NONE,
        CU_TENSOR_MAP_SWIZZLE_NONE,