#define VMA_IMPLEMENTATION
#include <ATen/native/vulkan/api/Allocator.h>