diff --git a/aiter/ops/triton/configs/gfx950/triton/attention/unified_attention/DEFAULT.json b/aiter/ops/triton/configs/gfx950/triton/attention/unified_attention/DEFAULT.json index 658f90ac639..f7285bc9383 100644 --- a/aiter/ops/triton/configs/gfx950/triton/attention/unified_attention/DEFAULT.json +++ b/aiter/ops/triton/configs/gfx950/triton/attention/unified_attention/DEFAULT.json @@ -132,6 +132,31 @@ "TILE_SIZE_MAX": 64, "SPLIT_UNMASKED_LOOP": true }, + "D_GEQ_256.Q_GEQ_256.SW.DT_fp8_fp8": { + "BLOCK_M": 128, + "num_warps": 4, + "num_stages": 1, + "waves_per_eu": 2, + "TILE_SIZE_MIN": 64, + "TILE_SIZE_MAX": 64 + }, + "D_GEQ_256.Q_GEQ_256.SHUF.DT_fp8_fp8": { + "BLOCK_M": 128, + "num_warps": 4, + "num_stages": 1, + "waves_per_eu": 2, + "TILE_SIZE_MIN": 64, + "TILE_SIZE_MAX": 64 + }, + "D_GEQ_256.Q_GEQ_256.DT_fp8_fp8": { + "BLOCK_M": 128, + "num_warps": 4, + "num_stages": 1, + "waves_per_eu": 2, + "TILE_SIZE_MIN": 64, + "TILE_SIZE_MAX": 64, + "SPLIT_UNMASKED_LOOP": true + }, "Q_GEQ_256": { "BLOCK_M": 128, "num_warps": 4,