auto-round
de2eca03 - fix: work around triton SPV_INTEL_predicated_io on Intel XPU

Commit
9 days ago
fix: work around triton SPV_INTEL_predicated_io on Intel XPU triton 3.7.x enables has_predicated_io for non-LTS Intel GPU drivers and emits the SPV_INTEL_predicated_io SPIR-V extension. Some level-zero loaders (e.g. 1.13.35563) reject that extension at kernel load time, aborting every triton kernel launch with 'InvalidModule: ... unknown extension'. Since it is not a catchable Python exception, the torch preprocess fallback never ran and the whole benchmark died. Add auto_round_kernel/_xpu_triton_compat.py which patches the intel triton backend to force has_predicated_io off (standard predicated loads instead), and apply it when the triton preprocess module loads. The triton-xpu preprocess backend now works and is faster than torch (seq 32768/50% e2e: 327ms -> 206ms).
Author
Committer
Parents
Loading