This is also necessary for https://github.com/intel/intel-xpu-backend-for-triton/pull/2484 to work correctly.