[ROCm] Retune MI355 selective_state_update float16 config on the unified effective_batch grid (#48372)

Signed-off-by: vanshbhatia-amd <210711135+vanshbhatia-amd@users.noreply.github.com>
Co-authored-by: vanshbhatia-amd <210711135+vanshbhatia-amd@users.noreply.github.com>

Author: vanshbhatia-amd

vllm/model_executor/layers/mamba/ops/configs/selective_state_update/headdim=64,dstate=128,device_name=AMD_Instinct_MI355_OAM,cache_dtype=float16.json

@@ -1,35 +1,51 @@
 {
     "triton_version": "3.6.0",
-    "10": {
+    "128": {
         "BLOCK_SIZE_M": 16,
         "num_warps": 4
     },
-    "80": {
-        "BLOCK_SIZE_M": 16,
+    "256": {
+        "BLOCK_SIZE_M": 32,
         "num_warps": 4
     },
-    "160": {
-        "BLOCK_SIZE_M": 16,
-        "num_warps": 4
+    "1024": {
+        "BLOCK_SIZE_M": 8,
+        "num_warps": 1
     },
-    "320": {
+    "2048": {
         "BLOCK_SIZE_M": 16,
-        "num_warps": 4
+        "num_warps": 2
     },
-    "640": {
-        "BLOCK_SIZE_M": 32,
-        "num_warps": 4
-    },
-    "1280": {
+    "4096": {
         "BLOCK_SIZE_M": 16,
         "num_warps": 1
     },
-    "2560": {
+    "8192": {
         "BLOCK_SIZE_M": 16,
         "num_warps": 1
     },
-    "5120": {
+    "16384": {
         "BLOCK_SIZE_M": 16,
-        "num_warps": 1
+        "num_warps": 2
+    },
+    "32768": {
+        "BLOCK_SIZE_M": 16,
+        "num_warps": 2
+    },
+    "65536": {
+        "BLOCK_SIZE_M": 16,
+        "num_warps": 2
+    },
+    "131072": {
+        "BLOCK_SIZE_M": 64,
+        "num_warps": 2
+    },
+    "196608": {
+        "BLOCK_SIZE_M": 16,
+        "num_warps": 2
+    },
+    "262144": {
+        "BLOCK_SIZE_M": 16,
+        "num_warps": 2
     }
 }