diff --git a/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td b/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td index 8fd533db83d9a..b143cf9a5f509 100644 --- a/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td +++ b/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td @@ -4196,7 +4196,8 @@ def SPIRV_Composite : def SPIRV_Type : AnyTypeOf<[ SPIRV_Void, SPIRV_Bool, SPIRV_Integer, SPIRV_Float, SPIRV_Vector, SPIRV_AnyPtr, SPIRV_AnyArray, SPIRV_AnyRTArray, SPIRV_AnyStruct, - SPIRV_AnyCooperativeMatrix, SPIRV_AnyMatrix, SPIRV_AnySampledImage + SPIRV_AnyCooperativeMatrix, SPIRV_AnyMatrix, SPIRV_AnySampledImage, + SPIRV_AnyImage ]>; def SPIRV_SignedInt : SignedIntOfWidths<[8, 16, 32, 64]>; diff --git a/mlir/test/Dialect/SPIRV/IR/memory-ops.mlir b/mlir/test/Dialect/SPIRV/IR/memory-ops.mlir index 57ff94762ff68..a3b96c698a344 100644 --- a/mlir/test/Dialect/SPIRV/IR/memory-ops.mlir +++ b/mlir/test/Dialect/SPIRV/IR/memory-ops.mlir @@ -356,6 +356,16 @@ spirv.module Logical GLSL450 { // ----- +// CHECK-LABEL: @image_load +func.func @image_load() -> () { + %0 = spirv.Variable : !spirv.ptr, Function> + // CHECK: spirv.Load "Function" %{{.*}} : !spirv.image + %1 = spirv.Load "Function" %0 : !spirv.image + return +} + +// ----- + //===----------------------------------------------------------------------===// // spirv.StoreOp //===----------------------------------------------------------------------===//