Skip to content

Commit 56d9a4f

Browse files
committed
Update base for Update on "[ET-VK][AOT][ez] Introduce vulkan export utils lib"
## Changes As title. Introduce a common Python utility library for scripts in the Vulkan backend. Differential Revision: [D65291064](https://our.internmc.facebook.com/intern/diff/D65291064/) [ghstack-poisoned]
1 parent 158e487 commit 56d9a4f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/vulkan/_passes/insert_prepack_nodes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,9 @@ def prepack_not_required(node: torch.fx.Node) -> bool:
7373
)
7474
# This pass assumes that the SpecPropPass() has already been applied
7575
assert "spec" in node.meta
76-
assert node.meta["spec"].const
7776
# Validate that the original node is marked as a constant. Constant tensors
7877
# do not participate in memory planning.
78+
assert node.meta["spec"].const
7979
prepack_node.meta["val"] = node.meta["val"]
8080
prepack_node.meta["spec"] = deepcopy(node.meta["spec"])
8181
# Set the mem_obj_id to -1 to indicate that this node requires a dedicated

0 commit comments

Comments
 (0)