mirror of https://github.com/llvm/torch-mlir
c69e9fabc5
This cleans up the lowering pipeline to easily allow extending to multiple binary ops. It looks fairly repetitive at multiple levels, but I don't want to prematurely generalize. I think that in principle we could derive a large swatch of TCF + TCP from a single linalg-style specification. Another direction is to use an OpInterface (something like "buildLinalgGenericBody"). I'm keeping my eye on it. In a subsequent commit, I'll mechanically add a set of binary ops modeled off of the std arithmetic ops. |
||
|---|---|---|
| .. | ||
| basic.mlir | ||
| binary-elementwise.mlir | ||
| broadcast.mlir | ||
| constant-add-scalar.mlir | ||
| constant-add.mlir | ||
| constant.mlir | ||
| control-flow-basic.mlir | ||
| identity.mlir | ||
| invalid-broadcast.mlir | ||
| invalid-matmul.mlir | ||
| invalid-num-inputs.mlir | ||
| matmul.mlir | ||
| mixed-rank.mlir | ||
| multi-output.mlir | ||
| multiple-ops.mlir | ||
| scalar.mlir | ||