diff --git a/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td b/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td --- a/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td +++ b/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td @@ -1208,19 +1208,22 @@ ]; } def SPV_C_AtomicFloat32AddEXT : I32EnumAttrCase<"AtomicFloat32AddEXT", 6033> { - list implies = [SPV_C_Shader]; + // Float atomics also supported in kernels. + // list implies = [SPV_C_Shader]; list availability = [ Extension<[SPV_EXT_shader_atomic_float_add]> ]; } def SPV_C_AtomicFloat64AddEXT : I32EnumAttrCase<"AtomicFloat64AddEXT", 6034> { - list implies = [SPV_C_Shader]; + // Float atomics also supported in kernels. + // list implies = [SPV_C_Shader]; list availability = [ Extension<[SPV_EXT_shader_atomic_float_add]> ]; } def SPV_C_AtomicFloat16AddEXT : I32EnumAttrCase<"AtomicFloat16AddEXT", 6095> { - list implies = [SPV_C_Shader]; + // Float atomics also supported in kernels. + // list implies = [SPV_C_Shader]; list availability = [ Extension<[SPV_EXT_shader_atomic_float16_add]> ];