From 918b1bb1f5f87faefbda90efb7a28430cb8ab995 Mon Sep 17 00:00:00 2001
From: Sicheng Stephen Jia <ssjia@meta.com>
Date: Wed, 11 Mar 2026 15:54:01 -0400
Subject: [PATCH] [ET-VK] Fix VMA assertion caused by bool return type
 truncation

Differential Revision: D96151798

Pull Request resolved: https://github.com/pytorch/executorch/pull/18105

(cherry picked from commit 7d43a9f3ef3f7e89700e72d60ea7e9087706fdf6)
---
 backends/vulkan/runtime/vk_api/memory/Allocator.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/backends/vulkan/runtime/vk_api/memory/Allocator.cpp b/backends/vulkan/runtime/vk_api/memory/Allocator.cpp
index b3c1a56529a..f5abf3e6d0c 100644
--- a/backends/vulkan/runtime/vk_api/memory/Allocator.cpp
+++ b/backends/vulkan/runtime/vk_api/memory/Allocator.cpp
@@ -11,7 +11,8 @@
 namespace vkcompute {
 namespace vkapi {
 
-bool test_host_cached_available(VkPhysicalDevice physical_device) {
+VmaAllocationCreateFlags test_host_cached_available(
+    VkPhysicalDevice physical_device) {
   VkPhysicalDeviceMemoryProperties mem_props;
   vkGetPhysicalDeviceMemoryProperties(physical_device, &mem_props);