Reland 60605d81b157b0b0df1cd0a7d9909932d22843a7 that is temporarily reverted in https://github.com/intel/intel-xpu-backend-for-triton/pull/5260 by f21b34136f063f791c6714d2ee6cdd1c645b7e83. CI: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/18328719993/job/52200044692 Related to #5074.