Skip to content

Commit

Permalink
Back out "tolist() support for FunctionalTensor"
Browse files Browse the repository at this point in the history
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
  • Loading branch information
jeanschmidt authored and facebook-github-bot committed Feb 26, 2024
1 parent 3b91dd2 commit bef80ce
Showing 1 changed file with 0 additions and 15 deletions.
15 changes: 0 additions & 15 deletions torchrec/distributed/tests/test_pt2.py
Original file line number Diff line number Diff line change
Expand Up @@ -240,18 +240,3 @@ def test_maybe_compute_kjt_to_jt_dict(self) -> None:
# TODO: turn on AOT Inductor test once the support is ready
test_aot_inductor=False,
)

def test_tensor_tolist(self) -> None:
class M(torch.nn.Module):
def forward(self, kjt: KeyedJaggedTensor):
return kjt.values().tolist()

kjt: KeyedJaggedTensor = make_kjt([2, 3, 4, 5, 6], [1, 2, 1, 1])
self._test_kjt_input_module(
M(),
kjt.keys(),
(kjt._values, kjt._lengths),
test_dynamo=False,
test_aot_inductor=False,
test_pt2_ir_export=True,
)

0 comments on commit bef80ce

Please sign in to comment.