UT added in https://github.com/intel/intel-xpu-backend-for-triton/pull/2978 from upstream commit https://github.com/triton-lang/triton/commit/5700c1468773d224075597f53710a79a796d5fd2.