xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/vk_api.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 
3 #ifdef USE_VULKAN_API
4 
5 #ifdef USE_VULKAN_WRAPPER
6 #ifdef USE_VULKAN_VOLK
7 #include <volk.h>
8 #else
9 #include <vulkan_wrapper.h>
10 #endif /* USE_VULKAN_VOLK */
11 #else
12 #include <vulkan/vulkan.h>
13 #endif /* USE_VULKAN_WRAPPER */
14 
15 #endif /* USE_VULKAN_API */
16