[FA]: squeeze Z and H into one axis to align with XeTLA#2618

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

quintin/index_alignintel/intel-xpu-backend-for-triton:quintin/index_alignCopy head branch name to clipboard