We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ae902e4 commit d96bb79Copy full SHA for d96bb79
userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -3203,3 +3203,9 @@ def get_user_object_from_id(obj_id):
3203
def store_user_object_weakref(obj):
3204
obj_id = id(obj)
3205
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