Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions backends/vulkan/partitioner/vulkan_partitioner.py
Original file line number Diff line number Diff line change
Expand Up @@ -251,6 +251,10 @@ def parse_compile_options(compile_options: Dict[str, Any]) -> List[CompileSpec]:
if isinstance(value, (VkStorageType, VkMemoryLayout)):
value_bytes = int(value).to_bytes(4, byteorder="little")
compile_specs.append(CompileSpec(key, value_bytes))

if isinstance(value, bool):
value_bytes = value.to_bytes(1, byteorder="little")
compile_specs.append(CompileSpec(key, value_bytes))

if key == "texture_limits":
compile_specs.append(
Expand Down
4 changes: 4 additions & 0 deletions backends/vulkan/vulkan_preprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,10 @@ def parse_compile_spec(compile_specs: List[CompileSpec]) -> Dict[str, Any]:
)
if spec.key in {"texture_limits_x", "texture_limits_y", "texture_limits_z"}:
options[spec.key] = int.from_bytes(spec.value, byteorder="little")

if spec.key == "skip_tag_memory_metadata":
options[spec.key] = bool.from_bytes(spec.value, byteorder="little")

# Unhandled options are ignored

return options
Expand Down
Loading