Skip to content

Commit

Permalink
Fix group fusion stride layout
Browse files Browse the repository at this point in the history
Summary:
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 27, 2024
1 parent ae902e4 commit d218f8b
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 @@ -3203,3 +3203,9 @@ def get_user_object_from_id(obj_id):
def store_user_object_weakref(obj):
obj_id = id(obj)
user_obj_id_to_weakref[obj_id] = weakref.ref(obj)


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 d218f8b

Please sign in to comment.