Skip to content

Commit 049c75e

Browse files
committed
Update base for 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]
1 parent 5fa37c1 commit 049c75e

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

backends/vulkan/test/vulkan_compute_api_test.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1660,9 +1660,8 @@ TEST(VulkanComputeGraphTest, test_simple_shared_objects_with_resize) {
16601660
for (auto& new_sizes : new_sizes_list) {
16611661
graph.get_tensor(a.value)->virtual_resize(new_sizes);
16621662
graph.get_tensor(b.value)->virtual_resize(new_sizes);
1663-
graph.get_tensor(c)->virtual_resize(new_sizes);
16641663
graph.get_tensor(d.value)->virtual_resize(new_sizes);
1665-
graph.get_tensor(e)->virtual_resize(new_sizes);
1664+
graph.propagate_resize();
16661665

16671666
float val_a = new_sizes[1] + 4.0f;
16681667
float val_b = new_sizes[2] + 1.5f;

0 commit comments

Comments
 (0)