Reland d36e57998e7aaf7495c5d916b51948b2a42ae5f3 that is temporarily reverted in https://github.com/intel/intel-xpu-backend-for-triton/pull/2783 by 016ee0055ccfc5a1340134747a2ad973699b9816.
