"...composable_kernel-1.git" did not exist on "b491ebf38480bc0d6cb329ba6825dee610c59097"
Commit 7409bc5d authored by ltqin's avatar ltqin
Browse files

add commits and fix maskdisabled

parent 57b856b3
...@@ -75,7 +75,7 @@ static constexpr auto MaskingSpec = ...@@ -75,7 +75,7 @@ static constexpr auto MaskingSpec =
ck::tensor_operation::device::MaskingSpecialization::MaskOutUpperTriangle; ck::tensor_operation::device::MaskingSpecialization::MaskOutUpperTriangle;
#else #else
static constexpr auto MaskingSpec = static constexpr auto MaskingSpec =
ck::tensor_operation::device::MaskingSpecialization::MaskOutUpperTriangle; ck::tensor_operation::device::MaskingSpecialization::MaskDisabled;
#endif #endif
static constexpr auto TensorSpecQ = ck::tensor_operation::device::TensorSpecialization::Default; static constexpr auto TensorSpecQ = ck::tensor_operation::device::TensorSpecialization::Default;
......
...@@ -1267,7 +1267,8 @@ struct GridwiseBatchedMultiheadAttentionBackward_Xdl_CShuffle ...@@ -1267,7 +1267,8 @@ struct GridwiseBatchedMultiheadAttentionBackward_Xdl_CShuffle
k_grid_desc_n0_k_n1, k_grid_desc_n0_k_n1,
make_multi_index(0, o_block_data_idx_on_grid, 0), make_multi_index(0, o_block_data_idx_on_grid, 0),
b1_element_op, b1_element_op,
Gemm1::b_block_desc_bk0_n_bk1, Gemm1::b_block_desc_bk0_n_bk1, // there n actually is k, k is N, so name can be
// b_block_desc_bn0_k_bn1
make_multi_index(0, 0, 0), make_multi_index(0, 0, 0),
tensor_operation::element_wise::PassThrough{}); tensor_operation::element_wise::PassThrough{});
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment