Skip to content

Commit

Permalink
Fix group fusion stride layout (pytorch#2442)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#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
  • Loading branch information
mengluy0125 authored and facebook-github-bot committed Aug 28, 2024
1 parent d873514 commit 0bc66e5
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -3242,3 +3242,9 @@ def record(cls):
finally:
if config.record_compile_time_instruction_count:
cls.end()


def realize_inputs(inputs: List[torch.fx.Node]):
for inp in inputs:
if isinstance(inp, torch.fx.node.Node):
inp.meta["inductor_realize_to_strides"] = True

0 comments on commit 0bc66e5

Please sign in to comment.