[Mlir-commits] [mlir] [mlir][AMDGPU] Add scaled floating point conversion ops fp8 (PR #141554)
Umang Yadav
llvmlistbot at llvm.org
Tue Jun 10 06:00:09 PDT 2025
================
@@ -112,6 +112,38 @@ def AMDGPU_ExtPackedFp8Op :
}];
}
+def AMDGPU_ScaledExtPackedOp
+ : AMDGPU_Op<"scaled_ext_packed", [Pure]>,
+ Arguments<(
+ ins AnyTypeOf<[VectorOfLengthAndType<[2, 3, 4], [F8E5M2, F8E4M3FN]>,
+ VectorOfLengthAndType<[2, 3, 4, 5, 6, 7, 8],
+ [F4E2M1FN]>]>:$source,
+ F32:$scale,
+ ConfinedAttr<I32Attr, [IntNonNegative, IntMaxValue<7>]>:$index)>,
+ Results<(
+ outs AnyTypeOf<[FixedVectorOfLengthAndType<[2], [F32]>,
+ FixedVectorOfLengthAndType<[2], [F16]>,
+ FixedVectorOfLengthAndType<[2], [BF16]>]>:$res)> {
+ let summary = "Extend a vector of packed floating point values";
+
+ let description = [{
+ Extend and scale two packed floats in `source[index]` to two floats and
+ return them.
----------------
umangyadav wrote:
I find this a bit confusing `source[index]` would only point to one element. How do we get two floats in return ?
https://github.com/llvm/llvm-project/pull/141554
More information about the Mlir-commits
mailing list