DeepSpeed

3a917de5 - Pass expected grad dtype to register_z3_param in ZeRO-3 release test (#8063)

Pass expected grad dtype to register_z3_param in ZeRO-3 release test (#8063) #8038 added an argument to `register_z3_param`, but `test_deepcompile_z3_release.py` wasn't updated. This PR updates the test accordingly. Signed-off-by: Masahiro Tanaka <mtanaka@anyscale.com>