[Bugfix][ROCm][DSv4] Skip launch_pdl=True JIT warmup when PDL is unsupported (#56035)
Signed-off-by: jimmy-adams <41593649+jimmy-adams@users.noreply.github.com>
jimmy-adams committed
Sep 9, 2026 at 21:50 UTC
62f3bf58a504fa58040d099f6fdd1a6171c90728
2 files changed
+2
-2
vllm/models/common/ops/fused_qk_rmsnorm.py
+1
-1
@@ -126,7 +126,7 @@ class FusedQKVRMSNormKernel(VllmTritonJitKernel["FusedQKVRMSNormKernel.CompileKe
126
kv_in_stride=input_stride,
127
kv_out_stride=(input_stride, kv_size),
128
eps=float(hf_config.rms_norm_eps),
129
- launch_pdl=(False, True),
129
+ launch_pdl=current_platform.is_arch_support_pdl(),
130
)
131
132
def warmup_inputs(self, compile_key: CompileKey) -> dict[str, Any]:
vllm/models/deepseek_v4/common/ops/fused_inv_rope_fp8_quant.py
+1
-1
@@ -249,7 +249,7 @@ class FusedInvRopeFP8QuantKernel(
249
rope_dim=rope_dim,
250
quant_group_size=128,
251
tma_aligned_scales=capability.major >= 10,
252
- launch_pdl=(False, True),
252
+ launch_pdl=current_platform.is_arch_support_pdl(),
253
)
254
255
def warmup_inputs(self, compile_key: CompileKey) -> dict[str, Any]: