tensor.splat

24 views
Skip to first unread message

Do Po

unread,
Apr 16, 2024, 8:17:36 AM4/16/24
to iree-discuss
Hello,

The last version of iree uses stablehlo as input dialect.
With respect to this statement, what is the status of more classical core MLIR dialects, such as tensor. I was surprised to discover that operation tensor.splat (of the tensor core dialect) is not accepted. 

Regards,
Dumitru

Jacques Pienaar

unread,
Apr 16, 2024, 1:48:20 PM4/16/24
to Do Po, iree-discuss
Hey,

I'd consider it more like 1) model support is top priority and that dictates what ops are prioritized, 2) ML framework produced input dialects (like TOSA or StableHLO) coverage is then secondary (both of these are still changing though so any coverage numbers is slice in time), 3) everything else. LinAlg is the primary "consolidation" point codegen wise in IREE, so things that can be lowered to LinAlg are supported more readily. 

That means while upstream MLIR dialects are supported, it's mostly those that have LinAlg lowerings or are produced along path 1 or 2 above. Now tensor::splat, I couldn't see any codegen for it even upstream. If it had a lowering to linalg then it would just work (the only Tensor dialect op that seems to have lowering to LinAlg is Pad it would seem). Else patches are welcome given priority is mostly focused on ML inputs (e.g., welcome but not priority)

-- Jacques

--
You received this message because you are subscribed to the Google Groups "iree-discuss" group.
To unsubscribe from this group and stop receiving emails from it, send an email to iree-discuss...@googlegroups.com.
To view this discussion on the web visit https://groups.google.com/d/msgid/iree-discuss/03a7b538-cdf2-4ee9-ab0d-d72bf83866fdn%40googlegroups.com.
Reply all
Reply to author
Forward
0 new messages