[rls-v3.7] xe: jit: gemm: rearrange runtime dim checks#2493