github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

add vulkan vma alloc flag

Browse Source
This commit is contained in:
PhaneeshB
2023-07-26 17:50:24 +05:30
committed by Phaneesh Barwaria
parent ab57af43c1
commit d6759a852b
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
apps/stable_diffusion/src/utils/stable_args.py
View File

@@ -446,6 +446,13 @@ p.add_argument(
help="Flag for disabling vulkan validation layers when benchmarking.",
)
p.add_argument(
"--vulkan_vma_allocator",
default=True,
action=argparse.BooleanOptionalAction,
help="Flag for enabling / disabling vulkan Vulkan VMA Allocator.",
)
##############################################################################
# Misc. Debug and Optimization flags
##############################################################################

1
apps/stable_diffusion/src/utils/utils.py
View File

@@ -186,6 +186,7 @@ def set_iree_runtime_flags():
vulkan_runtime_flags = [
f"--vulkan_large_heap_block_size={args.vulkan_large_heap_block_size}",
f"--vulkan_validation_layers={'true' if args.vulkan_validation_layers else 'false'}",
f"--vulkan_vma_allocator={'true' if args.vulkan_vma_allocator else 'false'}",
]
if args.enable_rgp:
vulkan_runtime_flags += [