set saving check to dp_rank == 0 in case of different layouts.#2
Open
dmahan93 wants to merge 1 commit intocat-state:mainfrom
Open
set saving check to dp_rank == 0 in case of different layouts.#2dmahan93 wants to merge 1 commit intocat-state:mainfrom
dmahan93 wants to merge 1 commit intocat-state:mainfrom
