[DeepCompile] Fix KeyError on frozen parameters in ZeRO-3 (#8214)