Skip to content

Commit c8a3918

Browse files
authored
read meta["val"] from arg for place holders instead of generating a fake tensor using fake_tensor_mode.from_real_tensor
Differential Revision: D65120377 Pull Request resolved: #6550
1 parent 8be6327 commit c8a3918

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exir/pass_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -453,7 +453,7 @@ def on_attr(self, attr: ProxyValue) -> None:
453453
def placeholder(self, name: str, arg: Argument, meta: NodeMetadata) -> ProxyValue:
454454
arg_proxy = self.tracer.create_proxy("placeholder", name, (), {})
455455
arg_proxy.node.meta = meta.data
456-
self.tracer.set_metadata(arg_proxy.node, arg)
456+
arg_proxy.node.meta["val"] = arg
457457
return ProxyValue(arg, arg_proxy)
458458

459459
def call_operator(

0 commit comments

Comments
 (0)