Reland upstream commit 4734af3 #5464
Description
Reland 4734af3 that is temporarily reverted in #5460 by f1e5aad.
CI: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/19287863486/job/55152692885
{{ message }}
Reland upstream commit