Fix DeepCompile ZeRO-3 gathered parameter ownership (#8157)