torch-mlir

Commit Graph

Author	SHA1	Message	Date
Sean Silva	ef25428fe3	Add lowering from linalg to loops. This also adds a small pass to clean up the `dim` ops that linalg introduces. For now, it only has a trivial pattern that looks for a `tcp.alloc_memref(%shape)` op to get the shape as we currently have an invariant that all memrefs are the result of such ops. But eventually this will need to look through view ops and any other shape-ish stuff that linalg introduces as it lowers to loops, along with any slicing ops introduced by buffer allocation.	2020-05-11 18:54:52 -07:00
Sean Silva	53c17dbed9	"Finish" tensor -> memref conversion. There's a lot of details to flesh out here, but the basic approach seems promising (see comments in createE2ELoweringPipeline). This approach will be put to the test when we try to do our first fusions since that tickles some of the nasty phase ordering issues involved here. But we're not there yet.	2020-05-11 15:00:12 -07:00
Sean Silva	e29aef855b	Initial TCF/TCP E2E seed. Very much WIP. This is enough to get tcf.add down to approximately the "linalg.generic on buffers" level of abstraction. (but there are nuances)	2020-05-08 20:20:41 -07:00

Author

SHA1

Message

Date

Sean Silva

ef25428fe3

Add lowering from linalg to loops.

This also adds a small pass to clean up the `dim` ops that linalg
introduces. For now, it only has a trivial pattern that looks for a
`tcp.alloc_memref(%shape)` op to get the shape as we currently have an
invariant that all memrefs are the result of such ops.

But eventually this will need to look through view ops and any other
shape-ish stuff that linalg introduces as it lowers to loops, along with
any slicing ops introduced by buffer allocation.

2020-05-11 18:54:52 -07:00

Sean Silva

53c17dbed9

"Finish" tensor -> memref conversion.

There's a lot of details to flesh out here, but the basic approach seems
promising (see comments in createE2ELoweringPipeline).

This approach will be put to the test when we try to do our first
fusions since that tickles some of the nasty phase ordering issues
involved here.

But we're not there yet.

2020-05-11 15:00:12 -07:00

Sean Silva

e29aef855b

Initial TCF/TCP E2E seed.

Very much WIP.

This is enough to get tcf.add down to approximately the "linalg.generic
on buffers" level of abstraction. (but there are nuances)

2020-05-08 20:20:41 -07:00

3 Commits (ef25428fe368511c6f694503e0514ea0b716ef95)