perf(dpa4): batch the SO3/grid contractions over (D,F); keep use_amp through pt_expt assembly - #5960

perf(dpa4): batch the SO3/grid contractions over (D,F); keep use_amp through pt_expt assembly#5960

wanghan-iapcm wants to merge 19 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