fix is mask tag

b4514459 · ltqin · f25d369b · b4514459
Commit b4514459 authored Jul 25, 2023 by ltqin
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/device/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_v2.hpp ...ice/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_v2.hpp +1 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_v2.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_v2.hpp
@@ -701,7 +701,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_V2
        CShuffleBlockTransferScalarPerVector_NPerBlock,
        LoopSched,
        Transform::matrix_padder.PadN,
-        MaskingSpec == MaskingSpecialization::MaskOutUpperTriangle,
+        MaskingSpec != MaskingSpecialization::MaskDisabled,
        Deterministic>;

    // Argument