Reenable test_permute for CW (#1494)

Summary: Pull Request resolved: #1494 As titiled Reviewed By: IvanKobzarev Differential Revision: D50884093 fbshipit-source-id: 3c540d44ea3d81c78b531e9584ea513df7ae8c8f
pytorch · Nov 9, 2023 · 9606fdd · 9606fdd
1 parent 2bd6519
commit 9606fdd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torchrec/distributed/tests/test_infer_shardings.py b/torchrec/distributed/tests/test_infer_shardings.py
@@ -134,7 +134,6 @@ def test_rw(self, weight_dtype: torch.dtype) -> None:
     )
     @settings(max_examples=4, deadline=None)
     def test_cw(self, test_permute: bool, weight_dtype: torch.dtype) -> None:
-        test_permute = False
         num_embeddings = 64
         emb_dim = 512
         emb_dim_4 = emb_dim // 4