[FP8] Unsqueeze scale to make it compatible with torch.compile#6040

ver217 merged 1 commit intohpcaitech:mainhpcaitech/ColossalAI:mainfrom

GuangyaoZhang:fp8_commGuangyaoZhang/ColossalAI:fp8_commCopy head branch name to clipboard