Skip to content

Commit 02af009

Browse files
author
ssjia
committed
Update on "[ET-VK] Introduce TextureMetadata struct"
Title says it all! Introduce a utility struct `TextureMetadata` to make it easier to pass tensor metadata to compute shaders, as a direct equivalent to `BufferMetadata`. Differential Revision: [D84716457](https://our.internmc.facebook.com/intern/diff/D84716457/) [ghstack-poisoned]
2 parents 1835517 + ed58653 commit 02af009

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)