Skip to content

Commit 2d7d21d

Browse files
committed
[mlir][spirv]: Add Image to Vulkan Storage Class Map
Extend the "storage class" <-> "memory space" map for the Vulkan SPIR-V environment to include the Image class. 12 is chosen as the next available value in the MemRef memory space list. Signed-off-by: Jack Frankland <[email protected]>
1 parent c309abd commit 2d7d21d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mlir/lib/Conversion/MemRefToSPIRV/MapMemRefStorageClassPass.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,8 @@ using namespace mlir;
5959
MAP_FN(spirv::StorageClass::UniformConstant, 8) \
6060
MAP_FN(spirv::StorageClass::Input, 9) \
6161
MAP_FN(spirv::StorageClass::Output, 10) \
62-
MAP_FN(spirv::StorageClass::PhysicalStorageBuffer, 11)
62+
MAP_FN(spirv::StorageClass::PhysicalStorageBuffer, 11) \
63+
MAP_FN(spirv::StorageClass::Image, 12)
6364

6465
std::optional<spirv::StorageClass>
6566
spirv::mapMemorySpaceToVulkanStorageClass(Attribute memorySpaceAttr) {

0 commit comments

Comments
 (0)