minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[AMD][Fix] Fix regression in jit build error with FLUX.2-dev on gfx1250 (#38329)

Browse Source
This commit is contained in:
sogalin_codegen
2026-09-08 22:55:13 -07:00
committed by GitHub
parent 00a9028e87
commit 5998e9321c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/kernels/ops/diffusion/norm/flux2_gated_resnorm_jit.py
+6 -1
View File
@@ -15,8 +15,13 @@ _ALIGNMENT = 32
def _blackwell_or_newer(device: torch.device) -> bool: def _blackwell_or_newer(device: torch.device) -> bool:
# ROCm reports a capability derived from the gfx number, so gfx1250 comes
# back as (12, 5) and clears a bare major >= 10 test. The kernel is PTX, so
# require a CUDA build like rope/qwen_qkv_epilogue_jit.py does.
return ( return (
torch.cuda.is_available() and torch.cuda.get_device_capability(device)[0] >= 10 torch.version.cuda is not None
and torch.cuda.is_available()
and torch.cuda.get_device_capability(device)[0] >= 10
) )