Unverified Commit efb0de2c authored by jacky.cheng's avatar jacky.cheng Committed by GitHub
Browse files

Update wave-lang to 3.7.0 and unify Wave kernel buffer options (#10069)

parent 0f6ac5e2
......@@ -82,7 +82,7 @@ srt_hip = [
"sglang[runtime_common]",
"torch",
"petit_kernel==0.0.2",
"wave-lang==1.0.1",
"wave-lang==3.7.0",
]
# https://docs.sglang.ai/platforms/cpu_server.html
......
......@@ -64,8 +64,7 @@ def get_wave_kernel(
subs=hyperparams_0,
canonicalize=True,
run_bench=False,
use_buffer_load_ops=True,
use_buffer_store_ops=True,
use_buffer_ops=True,
waves_per_eu=2,
dynamic_symbols=dynamic_symbols_0,
wave_runtime=True,
......@@ -77,8 +76,7 @@ def get_wave_kernel(
subs=hyperparams_1,
canonicalize=True,
run_bench=False,
use_buffer_load_ops=False,
use_buffer_store_ops=False,
use_buffer_ops=False,
waves_per_eu=4,
dynamic_symbols=dynamic_symbols_1,
wave_runtime=True,
......
......@@ -67,11 +67,9 @@ def get_wave_kernel(
schedule=SchedulingType.NONE,
use_scheduling_barriers=False,
dynamic_symbols=dynamic_symbols,
use_buffer_load_ops=True,
use_buffer_store_ops=True,
use_buffer_ops=True,
waves_per_eu=2,
denorm_fp_math_f32="preserve-sign",
gpu_native_math_precision=True,
wave_runtime=True,
)
options = set_default_run_config(options)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment