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.
2 parents 7e86248 + 56d9a4f commit 3dbd6cbCopy full SHA for 3dbd6cb
backends/vulkan/_passes/insert_prepack_nodes.py
@@ -56,9 +56,9 @@ def prepack_not_required(node: torch.fx.Node) -> bool:
56
)
57
# This pass assumes that the SpecPropPass() has already been applied
58
assert "spec" in node.meta
59
- assert node.meta["spec"].const
60
# Validate that the original node is marked as a constant. Constant tensors
61
# do not participate in memory planning.
+ assert node.meta["spec"].const
62
prepack_node.meta["val"] = node.meta["val"]
63
prepack_node.meta["spec"] = deepcopy(node.meta["spec"])
64
# Set the mem_obj_id to -1 to indicate that this node requires a dedicated
0 commit comments