From 222eed0220c7c5e7418e73e1d203342575141822 Mon Sep 17 00:00:00 2001 From: Ivan Butygin Date: Tue, 1 Jul 2025 15:21:49 +0200 Subject: [PATCH] remove dyn symbols Signed-off-by: Ivan Butygin --- .../srt/layers/attention/wave_ops/decode_attention.py | 10 ++++------ .../srt/layers/attention/wave_ops/extend_attention.py | 2 -- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/python/sglang/srt/layers/attention/wave_ops/decode_attention.py b/python/sglang/srt/layers/attention/wave_ops/decode_attention.py index 0037a242b610..030a3a38a799 100644 --- a/python/sglang/srt/layers/attention/wave_ops/decode_attention.py +++ b/python/sglang/srt/layers/attention/wave_ops/decode_attention.py @@ -70,8 +70,8 @@ def get_wave_kernel( phase_1, hyperparams_0, hyperparams_1, - dynamic_symbols, - dynamic_symbols_map, + dynamic_symbols_0, + dynamic_symbols_1, ) = get_paged_decode_attention_kernels( shape, mfma_variant, @@ -91,8 +91,7 @@ def get_wave_kernel( use_buffer_load_ops=True, use_buffer_store_ops=True, waves_per_eu=2, - dynamic_symbols=dynamic_symbols, - dynamic_symbols_map=dynamic_symbols_map, + dynamic_symbols=dynamic_symbols_0, wave_runtime=True, ) options = set_default_run_config(options) @@ -105,8 +104,7 @@ def get_wave_kernel( use_buffer_load_ops=False, use_buffer_store_ops=False, waves_per_eu=4, - dynamic_symbols=dynamic_symbols, - dynamic_symbols_map=dynamic_symbols_map, + dynamic_symbols=dynamic_symbols_1, wave_runtime=True, ) options = set_default_run_config(options) diff --git a/python/sglang/srt/layers/attention/wave_ops/extend_attention.py b/python/sglang/srt/layers/attention/wave_ops/extend_attention.py index 55c4bb66dc9c..27c5aee6034f 100644 --- a/python/sglang/srt/layers/attention/wave_ops/extend_attention.py +++ b/python/sglang/srt/layers/attention/wave_ops/extend_attention.py @@ -68,7 +68,6 @@ def get_wave_kernel( extend_attention, hyperparams, dynamic_symbols, - dynamic_symbols_map, ) = get_extend_attention_kernel( shape, mfma_variant, @@ -94,7 +93,6 @@ def get_wave_kernel( schedule=SchedulingType.NONE, use_scheduling_barriers=False, dynamic_symbols=dynamic_symbols, - dynamic_symbols_map=dynamic_symbols_map, use_buffer_load_ops=True, use_buffer_store_ops=True, waves_per_eu=2,