Skip to content

Commit e868d0b

Browse files
authored
update waves_per_eu to 1 (#3356)
1 parent 591e751 commit e868d0b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/sglang/srt/layers/attention/triton_ops/extend_attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,7 @@ def extend_attention_fwd(
351351

352352
extra_kargs = {}
353353
if is_hip_:
354-
extra_kargs = {"waves_per_eu": 4, "matrix_instr_nonkdim": 16, "kpack": 2}
354+
extra_kargs = {"waves_per_eu": 1, "matrix_instr_nonkdim": 16, "kpack": 2}
355355

356356
_fwd_kernel[grid](
357357
q_extend,

0 commit comments

Comments
 (0)