Skip to content

Commit bef80ce

Browse files
jeanschmidtfacebook-github-bot
authored andcommitted
Back out "tolist() support for FunctionalTensor"
Summary: Original commit changeset: a226bef5de4c Original Phabricator Diff: D53731064 bypass-github-export-checks (Note: this ignores all push blocking failures!) Reviewed By: atalman Differential Revision: D54189311 fbshipit-source-id: 0475b621422a1620f606ef7e01be9bd12e499afb
1 parent 3b91dd2 commit bef80ce

File tree

1 file changed

+0
-15
lines changed

1 file changed

+0
-15
lines changed

torchrec/distributed/tests/test_pt2.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -240,18 +240,3 @@ def test_maybe_compute_kjt_to_jt_dict(self) -> None:
240240
# TODO: turn on AOT Inductor test once the support is ready
241241
test_aot_inductor=False,
242242
)
243-
244-
def test_tensor_tolist(self) -> None:
245-
class M(torch.nn.Module):
246-
def forward(self, kjt: KeyedJaggedTensor):
247-
return kjt.values().tolist()
248-
249-
kjt: KeyedJaggedTensor = make_kjt([2, 3, 4, 5, 6], [1, 2, 1, 1])
250-
self._test_kjt_input_module(
251-
M(),
252-
kjt.keys(),
253-
(kjt._values, kjt._lengths),
254-
test_dynamo=False,
255-
test_aot_inductor=False,
256-
test_pt2_ir_export=True,
257-
)

0 commit comments

Comments
 (0)