1 files changed, 3 insertions, 0 deletions
diff --git a/src/vulkan/transfer_image.h b/src/vulkan/transfer_image.h
index 2beb9d8..e68b88e 100644
--- a/src/vulkan/transfer_image.h
+++ b/src/vulkan/transfer_image.h
@@ -32,6 +32,7 @@ class TransferImage : public Resource {
   TransferImage(Device* device,
                 const Format& format,
                 VkImageAspectFlags aspect,
+                VkImageType image_type,
                 uint32_t x,
                 uint32_t y,
                 uint32_t z,
@@ -63,6 +64,8 @@ class TransferImage : public Resource {
                                         uint32_t* memory_type_index);
   VkBufferImageCopy CreateBufferImageCopy(uint32_t mip_level);
 
+  VkImageViewType GetImageViewType() const;
+
   /// An extra `VkBuffer` is used to facilitate the transfer of data from the
   /// host into the `VkImage` on the device.
   VkBuffer host_accessible_buffer_ = VK_NULL_HANDLE;