change enum to MaskUpperTringleFrom

321b6c8e · ltqin · b4514459 · 321b6c8e
Commit 321b6c8e authored Jul 25, 2023 by ltqin
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

test/batched_gemm_softmax_gemm_permute/test_batched_gemm_softmax_gemm_permute_util.hpp ...m_permute/test_batched_gemm_softmax_gemm_permute_util.hpp +2 -2

No files found.
--- a/test/batched_gemm_softmax_gemm_permute/test_batched_gemm_softmax_gemm_permute_util.hpp
+++ b/test/batched_gemm_softmax_gemm_permute/test_batched_gemm_softmax_gemm_permute_util.hpp
@@ -174,7 +174,7 @@ struct DeviceInstanceWrapper_G2M1N1K1O1_TNTT_FP16_M128_N128_K32_O128
            2,              // CShuffleNXdlPerWavePerShuffle
            S<1, 32, 1, 8>, // CShuffleBlockTransferClusterLengths_MBlock_MPerBlock_NBlock_NPerBlock
            8,              // CShuffleBlockTransferScalarPerVector_NPerBlock
-            MaskingSpecialization::MaskOutUpperTriangle>; // MaskOutUpperTriangle
+            MaskingSpecialization::MaskUpperTringleFromTopLeft>; // MaskUpperTringleFromTopLeft

    bool IsSupported(int M, int N, int K, int O)
    {
@@ -315,7 +315,7 @@ struct DeviceInstanceWrapper_G2M1N1K1O1_TNTT_BF16_M128_N128_K32_O128
            2,              // CShuffleNXdlPerWavePerShuffle
            S<1, 32, 1, 8>, // CShuffleBlockTransferClusterLengths_MBlock_MPerBlock_NBlock_NPerBlock
            8,              // CShuffleBlockTransferScalarPerVector_NPerBlock
-            MaskingSpecialization::MaskOutUpperTriangle>; // MaskOutUpperTriangle
+            MaskingSpecialization::MaskUpperTringleFromTopLeft>; // MaskUpperTringleFromTopLeft

    bool IsSupported(int M, int N, int K, int O)
    {