Skip to content

Commit 878b2b5

Browse files
mengluy0125facebook-github-bot
authored andcommitted
Fix group fusion stride layout (#2442)
Summary: X-link: pytorch/pytorch#134696 Pull Request resolved: #2442 context: https://fb.workplace.com/groups/1075192433118967/permalink/1401282167176657/ moving the changes to the group gemm op has compilation errors, see details in D55606636 Differential Revision: D61888433
1 parent e6251f1 commit 878b2b5

File tree

1 file changed

+6
-0
lines changed
  • userbenchmark/dynamo/dynamobench/_dynamo

1 file changed

+6
-0
lines changed

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3242,3 +3242,9 @@ def record(cls):
32423242
finally:
32433243
if config.record_compile_time_instruction_count:
32443244
cls.end()
3245+
3246+
3247+
def realize_inputs(inputs: List[torch.fx.Node]):
3248+
for inp in inputs:
3249+
if isinstance(inp, torch.fx.node.Node):
3250+
inp.meta["inductor_realize_to_strides"] = True

0 commit comments

Comments
 (0)