Skip to content

GitLab

"client/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "db81d81b23053a8eaa4737c5704b9769bb0c2e2d"
Commit 08e2ae8d authored by danyao12's avatar danyao12
Browse files

modify Y_M1

parent 210aa44d
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
include/ck/tensor_operation/gpu/device/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_light_v1_protro.hpp
View file @ 08e2ae8d
...@@ -359,7 +359,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_Light_V1 ...@@ -359,7 +359,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_Light_V1
static constexpr index_t V_O1 = BK1; static constexpr index_t V_O1 = BK1;
static constexpr index_t Y_O1 = AK1; static constexpr index_t Y_O1 = AK1;
static constexpr index_t Y_M1 = AK1; static constexpr index_t Y_M1 = B1K1;
static constexpr auto padder = GemmGemmPadder<GemmSpec, static constexpr auto padder = GemmGemmPadder<GemmSpec,
Number<MPerBlock>, Number<MPerBlock>,
......
include/ck/tensor_operation/gpu/device/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_light_v2_protro.hpp
View file @ 08e2ae8d
...@@ -366,7 +366,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_Light_V2 ...@@ -366,7 +366,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_Light_V2
static constexpr index_t V_O1 = BK1; static constexpr index_t V_O1 = BK1;
static constexpr index_t Y_O1 = AK1; static constexpr index_t Y_O1 = AK1;
static constexpr index_t Y_M1 = AK1; static constexpr index_t Y_M1 = B1K1;
static constexpr auto padder = GemmGemmPadder<GemmSpec, static constexpr auto padder = GemmGemmPadder<GemmSpec,
Number<MPerBlock>, Number<MPerBlock>,
......
include/ck/tensor_operation/gpu/device/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_v1_protro.hpp
View file @ 08e2ae8d
...@@ -306,7 +306,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_V1 ...@@ -306,7 +306,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_V1
static constexpr index_t V_O1 = BK1; static constexpr index_t V_O1 = BK1;
static constexpr index_t Y_O1 = AK1; static constexpr index_t Y_O1 = AK1;
static constexpr index_t Y_M1 = AK1; static constexpr index_t Y_M1 = B1K1;
static constexpr auto padder = GemmGemmPadder<GemmSpec, static constexpr auto padder = GemmGemmPadder<GemmSpec,
Number<MPerBlock>, Number<MPerBlock>,
......
include/ck/tensor_operation/gpu/device/impl/device_batched_mha_bwd_xdl_cshuffle_qloop_v2_protro.hpp
View file @ 08e2ae8d
...@@ -313,7 +313,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_V2 ...@@ -313,7 +313,7 @@ struct DeviceBatchedMultiheadAttentionBackward_Qloop_Xdl_CShuffle_V2
static constexpr index_t V_O1 = BK1; static constexpr index_t V_O1 = BK1;
static constexpr index_t Y_O1 = AK1; static constexpr index_t Y_O1 = AK1;
static constexpr index_t Y_M1 = AK1; static constexpr index_t Y_M1 = B1K1;
static constexpr auto padder = GemmGemmPadder<GemmSpec, static constexpr auto padder = GemmGemmPadder<GemmSpec,
Number<MPerBlock>, Number<MPerBlock>,
......
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