Skip to content

Commit

Permalink
revert map/zip iterator related changes (#132528)
Browse files Browse the repository at this point in the history
Summary:
Need to revert due to internal hangs: S437700

This reverts commit b6c1490cc02316ffe85e5ae74651d80f0158ba64.

Revert "[dynamo] implement IteratorVariable and polyfill fallbacks for enumerate (#131725)"

This reverts commit 2576dbbc35d66e8e9ed6cb12216ccc424cb87ec3.

Revert "[dynamo] add itertools repeat/count bytecode reconstruction (#131716)"

This reverts commit 35b4de32fafc5ad024c20ef1275711bffc557ae9.

Revert "[dynamo] add lazy IteratorVariable implementations for map and zip (#131413)"

This reverts commit 7d282d87550787d8269593093519c2ad7c5032cd.

Fixes #ISSUE_NUMBER

X-link: pytorch/pytorch#132528
Approved by: https://github.com/ZainRizvi

Reviewed By: ZainRizvi

Differential Revision: D60747715

Pulled By: williamwen42

fbshipit-source-id: aae017a7926465908cccd64f1810050a80ee7ea6
  • Loading branch information
williamwen42 authored and facebook-github-bot committed Aug 5, 2024
1 parent 38880ca commit 342857c
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -1387,12 +1387,8 @@ def same(
"""Check correctness to see if ref and res match"""
if fp64_ref is None:
fp64_ref = ref
if isinstance(
ref, (list, tuple, collections.deque, torch.nn.ParameterList, torch.Size)
):
assert isinstance(
res, (list, tuple, collections.deque)
), f"type mismatch {type(ref)} {type(res)}"
if isinstance(ref, (list, tuple, torch.nn.ParameterList, torch.Size)):
assert isinstance(res, (list, tuple)), f"type mismatch {type(ref)} {type(res)}"
if len(ref) != len(res):
log_error("Length mismatch")
return False
Expand Down

0 comments on commit 342857c

Please sign in to comment.