@parametrize DTensor clipping to test clipping with inf norm, L1, and L2 norm (#3220)

Summary: Pull Request resolved: #3220

Reviewed By: tsunghsienlee

Differential Revision: D78706038

fbshipit-source-id: 32196c738e46bd67a67fe47f7d144f18f5a6d5f0

Author: wz337

torchrec/optim/tests/test_clipping.py

@@ -251,7 +251,7 @@ def _get_params_to_pg(
         return {param: [param.device_mesh.get_group()] for param in params}
 
     @with_comms
-    @parametrize("norm_type", ("inf",))
+    @parametrize("norm_type", ("inf", 1, 2))
     def test_dtensor_clip_all_gradients_norm(
         self, norm_type: Union[float, str]
     ) -> None: