[Kernel] Make SelfAttention prepared for AMX_FP16; More balanced task split in Cross Attention#466

pujiang2018 merged 2 commits intomainintel/xFasterTransformer:mainfrom

pujiang/feature/amx_fp16intel/xFasterTransformer:pujiang/feature/amx_fp16Copy head branch name to clipboard