diff --git a/tests/unit/test_tma_minimal.cu b/tests/unit/test_tma_minimal.cu
index b3fc0830..23ef00fb 100644
--- a/tests/unit/test_tma_minimal.cu
+++ b/tests/unit/test_tma_minimal.cu
@@ -18,7 +18,7 @@
 #endif
 
 #ifndef NUM_THREADS
-#define NUM_THREADS 32
+#define NUM_THREADS 128
 #endif
 
 typedef unsigned short bf16_t;