From 55d289c65bc7931455aa634ed0e96bf11e9234c6 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 16:51:40 +0000
Subject: [PATCH] Fix TMA: use CU_TENSOR_MAP_DATA_TYPE_BFLOAT16 not UINT16

---
 tests/unit/test_tma_driver.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/test_tma_driver.cu b/tests/unit/test_tma_driver.cu
index 3a48a7a1..d7f08c7b 100644
--- 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,