Skip to content

Commit f4a4916

Browse files
committed
Update on "[ET-VK][AOT] Define pass application order"
## Changes The goal of this diff is to enforce a specific structure in how graph transform passes are applied during `vulkan_preprocess`. This will help make sure that certain passes are applied at the correct time, and that pre-requisite conditions for passes are fulfilled before they are applied. See the comments in `vulkan_preprocess.py` for more details. Differential Revision: [D65234843](https://our.internmc.facebook.com/intern/diff/D65234843/) [ghstack-poisoned]
2 parents 0aeec92 + 70adc6d commit f4a4916

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)