[Mlir-commits] [mlir] [mlir][gpu] Add extra value types for gpu::ShuffleOp (PR #104605)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Mon Aug 19 08:57:19 PDT 2024
================
@@ -1250,17 +1250,17 @@ def GPU_ShuffleMode : I32EnumAttr<"ShuffleMode",
def GPU_ShuffleModeAttr : EnumAttr<GPU_Dialect, GPU_ShuffleMode,
"shuffle_mode">;
-def I32I64F32OrF64 : TypeConstraint<Or<[I32.predicate,
- I64.predicate,
- F32.predicate,
- F64.predicate]>,
- "i32, i64, f32 or f64">;
+def GPUShuffleTypes : TypeConstraint<Or<[
+ SignlessIntOfWidths<[8, 16, 32, 64]>.predicate,
+ FloatOfWidths<[16, 32, 64]>.predicate
----------------
FMarno wrote:
These are the types supported by the [OpenCL SPIR-V Environment specification](https://registry.khronos.org/OpenCL/specs/3.0-unified/html/OpenCL_Env.html#_cl_khr_subgroup_shuffle), which is what I am personally targetting.
https://github.com/llvm/llvm-project/pull/104605
More information about the Mlir-commits
mailing list