Reland upstream commit `9b75018`

[whitneywhtsang]

Reland 9b75018 that is temporarily reverted in #5522 by bccf48a.

CI: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/19547076240/job/55968560616

[slawblauciak]

PR created: #5594