perf(dpa4): stop broadcasting weights across the node axis; fix use_amp serialization - #5960

perf(dpa4): stop broadcasting weights across the node axis; fix use_amp serialization#5960

wanghan-iapcm wants to merge 11 commits into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

wanghan-iapcm:perf-dpa4-grid-contractwanghan-iapcm/deepmd-kit:perf-dpa4-grid-contractCopy head branch name to clipboard