From 9606fdd23ad8d5b9f05ea8f190cd6329f0696993 Mon Sep 17 00:00:00 2001 From: Qiang Zhang Date: Wed, 8 Nov 2023 20:20:04 -0800 Subject: [PATCH] Reenable test_permute for CW (#1494) Summary: Pull Request resolved: https://github.com/pytorch/torchrec/pull/1494 As titiled Reviewed By: IvanKobzarev Differential Revision: D50884093 fbshipit-source-id: 3c540d44ea3d81c78b531e9584ea513df7ae8c8f --- torchrec/distributed/tests/test_infer_shardings.py | 1 - 1 file changed, 1 deletion(-) diff --git a/torchrec/distributed/tests/test_infer_shardings.py b/torchrec/distributed/tests/test_infer_shardings.py index 4e4687b04..07eb0c19a 100755 --- 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