Skip to content

Commit b2fd947

Browse files
[executorch][qnn] Fix pass migration CI failures (pytorch#22020)
Authored with Codex. cc @cbilgin @psiddh
1 parent 90f88b7 commit b2fd947

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

‎backends/qualcomm/builders/utils.py‎

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@ def is_parameter(
2121
node: torch.fx.Node, edge_program: torch.export.ExportedProgram
2222
) -> bool:
2323
return (
24-
is_param(edge_program, node)
24+
node.op == "get_attr"
25+
or is_param(edge_program, node)
2526
or is_buffer(edge_program, node)
2627
or is_lifted_tensor_constant(edge_program, node)
2728
)
@@ -37,9 +38,15 @@ def get_parameter(
3738
param = get_buffer(edge_program, node)
3839
if is_lifted_tensor_constant(edge_program, node):
3940
param = get_lifted_tensor_constant(edge_program, node)
41+
if node.op == "get_attr":
42+
assert isinstance(node.target, str)
43+
try:
44+
param = getattr(node.graph.owning_module, node.target)
45+
except AttributeError:
46+
param = getattr(edge_program.graph_module, node.target)
4047
assert (
4148
param is not None
42-
), f"Expect {node.name} to be parameter, buffer, or lifted tensor constant"
49+
), f"Expect {node.name} to be parameter, buffer, get_attr, or lifted tensor constant"
4350
# update node.meta["val"] to qualified QNN datatype (e.g. i64 to i32)
4451
assert isinstance(param, torch.Tensor), "Expect parameter to be tensor"
4552
param = param.type(node.meta["val"].dtype)

0 commit comments

Comments
 (0)