diff --git a/lib/Transforms/SetSPIRVCapabilities.cpp b/lib/Transforms/SetSPIRVCapabilities.cpp index ab508c73a..e2bb3ce70 100644 --- a/lib/Transforms/SetSPIRVCapabilities.cpp +++ b/lib/Transforms/SetSPIRVCapabilities.cpp @@ -70,6 +70,7 @@ struct SetSPIRVCapabilitiesPass spirv::Capability::AtomicFloat32AddEXT, spirv::Capability::ExpectAssumeKHR, spirv::Capability::VectorAnyINTEL, + spirv::Capability::VectorComputeINTEL, // clang-format on }; spirv::Capability caps_vulkan[] = { diff --git a/test/Transforms/set-spirv-capability.mlir b/test/Transforms/set-spirv-capability.mlir index 954015b50..1c49e570a 100644 --- a/test/Transforms/set-spirv-capability.mlir +++ b/test/Transforms/set-spirv-capability.mlir @@ -4,7 +4,7 @@ module attributes {gpu.container_module} { // OPENCL: module attributes {gpu.container_module} { -// OPENCL: gpu.module @main_kernel attributes {spirv.target_env = #spirv.target_env<#spirv.vce, api=OpenCL, #spirv.resource_limits<>>} { +// OPENCL: gpu.module @main_kernel attributes {spirv.target_env = #spirv.target_env<#spirv.vce, api=OpenCL, #spirv.resource_limits<>>} { // VULKAN: module attributes {gpu.container_module} { // VULKAN: gpu.module @main_kernel attributes {spirv.target_env = #spirv.target_env<#spirv.vce, api=Vulkan, #spirv.resource_limits<>>} { gpu.module @main_kernel {