diff --git a/examples/models/qwen3_5_moe/export.py b/examples/models/qwen3_5_moe/export.py index 92408a7ea71..4c060ce0137 100644 --- a/examples/models/qwen3_5_moe/export.py +++ b/examples/models/qwen3_5_moe/export.py @@ -1138,9 +1138,8 @@ def _export_cuda(model, config, args): "enable_dynamic_shape": True, "get_mutable_buffer_metadata": _mutable_buffer_metadata_json(model), } - # Avoid PyTorch 2.13's fused cast/reduction compile-time autotune path. Keep - # max autotuning enabled so prefill GEMMs use the CUDA backend's Triton-only - # GEMM lowering instead of emitting libtorch fallback kernels. + # Keep max autotuning enabled so prefill GEMMs use the CUDA backend's + # Triton-only GEMM lowering instead of emitting libtorch fallback kernels. et_prog = to_edge_transform_and_lower( {"decode": decode_ep, "prefill": prefill_ep}, partitioner={ @@ -1151,7 +1150,6 @@ def _export_cuda(model, config, args): CompileSpec("low_memory_mode", b"ON"), CompileSpec("emulate_precision_casts", b"OFF"), CompileSpec("max_autotune", b"ON"), - CompileSpec("autotune_at_compile_time", b"OFF"), ] ) ], @@ -1162,7 +1160,6 @@ def _export_cuda(model, config, args): CompileSpec("low_memory_mode", b"ON"), CompileSpec("emulate_precision_casts", b"OFF"), CompileSpec("max_autotune", b"ON"), - CompileSpec("autotune_at_compile_time", b"OFF"), ] ) ],