Skip to content

Commit d96bb79

Browse files
mengluy0125facebook-github-bot
authored andcommitted
Fix group fusion for AFOC
Summary: X-link: pytorch/pytorch#122839 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: D55449814
1 parent ae902e4 commit d96bb79

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

+6
Original file line numberDiff line numberDiff line change
@@ -3203,3 +3203,9 @@ def get_user_object_from_id(obj_id):
32033203
def store_user_object_weakref(obj):
32043204
obj_id = id(obj)
32053205
user_obj_id_to_weakref[obj_id] = weakref.ref(obj)
3206+
3207+
3208+
def realize_inputs(inputs: List[torch.fx.Node]):
3209+
for inp in inputs:
3210+
if isinstance(inp, torch.fx.node.Node):
3211+
inp.meta["inductor_realize_to_strides"] = True

0 commit comments

Comments
 (0)