[Mlir-commits] [mlir] [mlir][tosa] Support `DenseResourceElementsAttr` in TOSA transpose folders (PR #124532)
Luke Hutton
llvmlistbot at llvm.org
Thu Feb 20 04:02:23 PST 2025
================
@@ -204,6 +228,35 @@ DenseElementsAttr transpose(ElementsAttr attr, ShapedType inputType,
if (auto data = attr.tryGetValues<APFloat>())
return transposeType(*data, inputType, outputType, permValues);
+ // Handle DenseResourceElementsAttr
+ if (isa<DenseResourceElementsAttr>(attr)) {
+ auto elementTy = attr.getElementType();
+
+ if (auto data = tryGetDenseResourceValues<bool>(attr);
+ data && elementTy.isInteger(1))
+ return transposeType(*data, inputType, outputType, permValues);
+
+ if (auto data = tryGetDenseResourceValues<int8_t>(attr);
+ data && elementTy.isInteger(8))
+ return transposeType(*data, inputType, outputType, permValues);
+
+ if (auto data = tryGetDenseResourceValues<int16_t>(attr);
+ data && elementTy.isInteger(16))
+ return transposeType(*data, inputType, outputType, permValues);
+
+ if (auto data = tryGetDenseResourceValues<int32_t>(attr);
+ data && elementTy.isInteger(32))
+ return transposeType(*data, inputType, outputType, permValues);
+
+ if (auto data = tryGetDenseResourceValues<int64_t>(attr);
+ data && elementTy.isInteger(64))
+ return transposeType(*data, inputType, outputType, permValues);
+
+ if (auto data = tryGetDenseResourceValues<float>(attr);
----------------
lhutton1 wrote:
Is it worth adding this as a TODO below?
https://github.com/llvm/llvm-project/pull/124532
More information about the Mlir-commits
mailing list