Reland upstream commit 60605d8 #5269
Description
Reland 60605d8 that is temporarily reverted in #5260 by f21b341.
CI: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/18328719993/job/52200044692
Related to #5074.
{{ message }}
Reland upstream commit