Skip to content

Commit ea06e9a

Browse files
author
ssjia
committed
Update on "[ET-VK][ez] Introduce a graph config setting to force resize functions to execute"
Title says it all! A few months back, a mechanism was introduced where an `ExecuteNode` would not call an operator's resize function if none of the arguments were updated. However, this creates a blind spot during testing where the resize function of operators are not tested since the generated operator tests do not modify input sizes. To address this, add a way to force the resize function to be called during testing. Differential Revision: [D84716451](https://our.internmc.facebook.com/intern/diff/D84716451/) [ghstack-poisoned]
2 parents 14ac9ed + 463ab43 commit ea06e9a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

backends/vulkan/test/tester.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,9 @@ def __init__(
4444

4545
class Partition(BaseStages.Partition):
4646
def __init__(self, partitioner: Optional[Partitioner] = None):
47+
vk_compile_spec = {"skip_bool_tensors": True}
4748
super().__init__(
48-
partitioner=partitioner or VulkanPartitioner(),
49+
partitioner=partitioner or VulkanPartitioner(vk_compile_spec),
4950
)
5051

5152

@@ -55,6 +56,10 @@ def __init__(
5556
partitioners: Optional[List[Partitioner]] = None,
5657
edge_compile_config: Optional[EdgeCompileConfig] = None,
5758
):
59+
if partitioners is None:
60+
vk_compile_spec = {"skip_bool_tensors": True}
61+
partitioners = [VulkanPartitioner(vk_compile_spec)]
62+
5863
super().__init__(
5964
default_partitioner_cls=VulkanPartitioner,
6065
partitioners=partitioners,

0 commit comments

Comments
 (0)