Skip to content

Commit 39e5744

Browse files
committed
Update on "[ET-VK][ez] Updates to DynamicDispatchNode"
## Changes * Pass in global work group size to the local work group size determination function in `DynamicDIspatchNode` ## Motivation Oftentimes it is useful to know what the global work group size is when determining what the local group group size should be. ## Performance Impact None. Differential Revision: [D75686047](https://our.internmc.facebook.com/intern/diff/D75686047/) [ghstack-poisoned]
2 parents c059407 + baca926 commit 39e5744

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

backends/vulkan/runtime/VulkanBackend.cpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -499,6 +499,10 @@ class VulkanBackend final : public ::executorch::runtime::BackendInterface {
499499
compute_graph->encode_prepack();
500500
compute_graph->prepack();
501501

502+
// TODO(ssjia): remove this once we can batch compile compute pipelines
503+
// during prepare().
504+
compute_graph->encode_execute();
505+
502506
return Error::Ok;
503507
}
504508

@@ -572,12 +576,7 @@ class VulkanBackend final : public ::executorch::runtime::BackendInterface {
572576
if (should_propagate_resize) {
573577
compute_graph->propagate_resize();
574578
}
575-
// If propagate_resize() was not triggered (i.e. command buffer has not been
576-
// encoded), then make sure to encode the command buffer before executing
577-
// the first inference of the model.
578-
else if (compute_graph->execute_count() == 0) {
579-
compute_graph->encode_execute();
580-
}
579+
581580
compute_graph->execute();
582581

583582
for (size_t i = 0; i < compute_graph->outputs().size(); i++) {

0 commit comments

Comments
 (0)