Skip to content

Commit 18dba91

Browse files
manuelcandalesfacebook-github-bot
authored andcommitted
Fix select_dtype_combinations return for GenMode.All (#339)
Summary: Pull Request resolved: #339 Reviewed By: larryliu0820 Differential Revision: D49217789 fbshipit-source-id: 959456fca892a60e14592c17eab3d7847433bc68
1 parent d1a009e commit 18dba91

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

exir/dialects/edge/dtype/runner.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,8 @@ def produce_code_tuples(n: int, i: int) -> Iterator[Tuple[int]]:
111111

112112
type_tuples = self._get_type_tuples(inputs)
113113
if genmode == GenMode.All:
114-
return itertools.product(*type_tuples) # noqa
114+
for dtype_tuple in itertools.product(*type_tuples):
115+
yield dtype_tuple
115116
elif genmode == GenMode.Partial:
116117
dtype_tuples_set = set()
117118
types = DtypeRunner._get_types(inputs)

0 commit comments

Comments
 (0)