[AMP] refine AMP and the corresponding tests for bfloat16#12787

masahi merged 3 commits into

apache:mainapache/tvm:mainfrom

yangulei:fix_ampyangulei/tvm:fix_ampCopy head branch name to clipboard