get rid of one small allgather
get rid of one small allgather

toxa81/sirius,toxa81/sirius,toxa81/sirius,electronic-structure/sirius,toxa81/sirius,electronic-structure/sirius,toxa81/sirius,toxa81/sirius,electronic-structure/sirius,electronic-structure/sirius,electronic-structure/sirius,electronic-structure/sirius,electronic-structure/sirius