From bfef94f5d0aed8d1b5719c581ea0353cecb82e97 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 29 May 2026 19:40:32 +0000 Subject: [PATCH] test: HD=128/256 multi-row TMA FMHA --- tests/unit/test_fmha_6warp_tma_multirow_hd128.cu | 2 ++ tests/unit/test_fmha_6warp_tma_multirow_hd256.cu | 2 ++ 2 files changed, 4 insertions(+) create mode 100644 tests/unit/test_fmha_6warp_tma_multirow_hd128.cu create mode 100644 tests/unit/test_fmha_6warp_tma_multirow_hd256.cu diff --git a/tests/unit/test_fmha_6warp_tma_multirow_hd128.cu b/tests/unit/test_fmha_6warp_tma_multirow_hd128.cu new file mode 100644 index 00000000..f72ac985 --- /dev/null +++ b/tests/unit/test_fmha_6warp_tma_multirow_hd128.cu @@ -0,0 +1,2 @@ +#define HD_VAL 128 +#include "test_fmha_6warp_tma_multirow.cu" diff --git a/tests/unit/test_fmha_6warp_tma_multirow_hd256.cu b/tests/unit/test_fmha_6warp_tma_multirow_hd256.cu new file mode 100644 index 00000000..b4a29c6a --- /dev/null +++ b/tests/unit/test_fmha_6warp_tma_multirow_hd256.cu @@ -0,0 +1,2 @@ +#define HD_VAL 256 +#include "test_fmha_6warp_tma_multirow.cu"