Skip to content

Commit 3dbd6cb

Browse files
committed
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]
2 parents 7e86248 + 56d9a4f commit 3dbd6cb

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
@@ -56,9 +56,9 @@ def prepack_not_required(node: torch.fx.Node) -> bool:
5656
)
5757
# This pass assumes that the SpecPropPass() has already been applied
5858
assert "spec" in node.meta
59-
assert node.meta["spec"].const
6059
# Validate that the original node is marked as a constant. Constant tensors
6160
# do not participate in memory planning.
61+
assert node.meta["spec"].const
6262
prepack_node.meta["val"] = node.meta["val"]
6363
prepack_node.meta["spec"] = deepcopy(node.meta["spec"])
6464
# Set the mem_obj_id to -1 to indicate that this node requires a dedicated

0 commit comments

Comments
 (0)