karylab_agents/vllm
1
0
Fork 0
forked from Karylab-cklius/vllm
Code Pull Requests 2 Actions 1 Packages Activity

[ROCm] Add tuned selective_state_update config for AMD MI350 (#48159)

Browse Source 
Signed-off-by: Giuseppe Grossi <ggrossi@amd.com>
This commit is contained in:
Giuseppe Grossi
2026-07-15 10:18:09 +03:00
committed by GitHub
parent 12a8057bfe
commit d119beb1b9
2 changed files with 102 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vllm/model_executor/layers/mamba/ops/configs/selective_state_update/headdim=64,dstate=128,device_name=AMD_Instinct_MI350_OAM,cache_dtype=float16.json
+51
View File
@@ -0,0 +1,51 @@
{
"triton_version": "3.6.0",
"128": {
"BLOCK_SIZE_M": 16,
"num_warps": 4
},
"256": {
"BLOCK_SIZE_M": 32,
"num_warps": 4
},
"1024": {
"BLOCK_SIZE_M": 16,
"num_warps": 2
},
"2048": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
},
"4096": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
},
"8192": {
"BLOCK_SIZE_M": 16,
"num_warps": 2
},
"16384": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
},
"32768": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
},
"65536": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
},
"131072": {
"BLOCK_SIZE_M": 16,
"num_warps": 1
},
"196608": {
"BLOCK_SIZE_M": 16,
"num_warps": 1
},
"262144": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
}
}
vllm/model_executor/layers/mamba/ops/configs/selective_state_update/headdim=64,dstate=128,device_name=AMD_Instinct_MI350_OAM,cache_dtype=float32.json
+51
View File
@@ -0,0 +1,51 @@
{
"triton_version": "3.6.0",
"128": {
"BLOCK_SIZE_M": 16,
"num_warps": 4
},
"256": {
"BLOCK_SIZE_M": 16,
"num_warps": 4
},
"1024": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
},
"2048": {
"BLOCK_SIZE_M": 8,
"num_warps": 1
},
"4096": {
"BLOCK_SIZE_M": 4,
"num_warps": 1
},
"8192": {
"BLOCK_SIZE_M": 32,
"num_warps": 2
},
"16384": {
"BLOCK_SIZE_M": 4,
"num_warps": 1
},
"32768": {
"BLOCK_SIZE_M": 4,
"num_warps": 1
},
"65536": {
"BLOCK_SIZE_M": 64,
"num_warps": 8
},
"131072": {
"BLOCK_SIZE_M": 64,
"num_warps": 8
},
"196608": {
"BLOCK_SIZE_M": 64,
"num_warps": 2
},
"262144": {
"BLOCK_SIZE_M": 64,
"num_warps": 8
}
}