Reland upstream commit `d16ef81`#2528

whitneywhtsang merged 2 commits intomainintel/intel-xpu-backend-for-triton:mainfrom

whitneywhtsang/merge3intel/intel-xpu-backend-for-triton:whitneywhtsang/merge3Copy head branch name to clipboard