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 158e487 commit 56d9a4fCopy full SHA for 56d9a4f
backends/vulkan/_passes/insert_prepack_nodes.py
@@ -73,9 +73,9 @@ def prepack_not_required(node: torch.fx.Node) -> bool:
73
)
74
# This pass assumes that the SpecPropPass() has already been applied
75
assert "spec" in node.meta
76
- assert node.meta["spec"].const
77
# Validate that the original node is marked as a constant. Constant tensors
78
# do not participate in memory planning.
+ assert node.meta["spec"].const
79
prepack_node.meta["val"] = node.meta["val"]
80
prepack_node.meta["spec"] = deepcopy(node.meta["spec"])
81
# Set the mem_obj_id to -1 to indicate that this node requires a dedicated
0 commit comments