Skip to content

GitLab

Unverified Commit 8a248427 authored by Amanzhol Salykov's avatar Amanzhol Salykov Committed by GitHub
Browse files

[ROCm] add tuned moe_wna16_triton kernel configs for CDNA4 (#35093) 


Signed-off-by: default avatarsalykova <amsalykov@gmail.com>
Signed-off-by: default avataramd-asalykov <asalykov@amd.com>
parent 65986db6
Hide whitespace changes
Inline Side-by-side
Showing with 768 additions and 0 deletions
vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=AMD_Instinct_MI350X,dtype=int4_w4a16.json 0 → 100644
View file @ 8a248427
{
"1": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"2": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"4": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"16": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"24": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"32": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"48": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"64": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"96": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"128": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"256": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"512": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1024": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1536": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"2048": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"3072": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"4096": {
"BLOCK_SIZE_M": 128,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8192": {
"BLOCK_SIZE_M": 256,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 32,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 8,
"num_stages": 2,
"matrix_instr_nonkdim": 32
}
}
\ No newline at end of file
vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=AMD_Instinct_MI350_OAM,dtype=int4_w4a16.json 0 → 100644
View file @ 8a248427
{
"1": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"2": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"4": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"16": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"24": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"32": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"48": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"64": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"96": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"128": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"256": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"512": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1024": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1536": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"2048": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"3072": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"4096": {
"BLOCK_SIZE_M": 128,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8192": {
"BLOCK_SIZE_M": 256,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 32,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 8,
"num_stages": 2,
"matrix_instr_nonkdim": 32
}
}
\ No newline at end of file
vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=AMD_Instinct_MI355X,dtype=int4_w4a16.json 0 → 100644
View file @ 8a248427
{
"1": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"2": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"4": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"16": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"24": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"32": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"48": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"64": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"96": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"128": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"256": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"512": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1024": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1536": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"2048": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"3072": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"4096": {
"BLOCK_SIZE_M": 128,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8192": {
"BLOCK_SIZE_M": 256,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 32,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 8,
"num_stages": 2,
"matrix_instr_nonkdim": 32
}
}
\ No newline at end of file
vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=AMD_Instinct_MI355_OAM,dtype=int4_w4a16.json 0 → 100644
View file @ 8a248427
{
"1": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"2": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"4": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"16": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"24": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"32": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"48": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"64": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"96": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"128": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"256": {
"BLOCK_SIZE_M": 16,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"512": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1024": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"1536": {
"BLOCK_SIZE_M": 32,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 32
},
"2048": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"3072": {
"BLOCK_SIZE_M": 64,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 128,
"GROUP_SIZE_M": 1,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 1,
"matrix_instr_nonkdim": 16
},
"4096": {
"BLOCK_SIZE_M": 128,
"BLOCK_SIZE_N": 128,
"BLOCK_SIZE_K": 64,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 4,
"num_stages": 2,
"matrix_instr_nonkdim": 16
},
"8192": {
"BLOCK_SIZE_M": 256,
"BLOCK_SIZE_N": 256,
"BLOCK_SIZE_K": 32,
"GROUP_SIZE_M": 2,
"SPLIT_K": 1,
"num_warps": 8,
"num_stages": 2,
"matrix_instr_nonkdim": 32
}
}
\ No newline at end of file
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