torch-mlir

Commit Graph

Author	SHA1	Message	Date
Sean Silva	32388d938b	Make some passes run on FuncOp so they can run in parallel.	2020-11-13 16:12:18 -08:00
Sean Silva	1c7c362e29	[TCP] Replace tcp.matmul with linalg.matmul. This involved adding a `tcp.splatted` op to splat a dynamically sized init tensor. See rationale in TCPOps.td docs. One interesting observation is that when lowering tcf.matmul to linalg.matmul, we need to both 1) create the error checks and 2) calculate a shape transfer function to create the init tensors. Previously, 2) was deferred to bufferizing tcp.matmul later. I'm not sure if this is a conflation of concerns or not. For now, it's not a big burden.	2020-11-10 18:58:28 -08:00

Author

SHA1

Message

Date

Sean Silva

32388d938b

Make some passes run on FuncOp so they can run in parallel.

2020-11-13 16:12:18 -08:00

Sean Silva

1c7c362e29

[TCP] Replace tcp.matmul with linalg.matmul.

This involved adding a `tcp.splatted` op to splat a dynamically sized
init tensor. See rationale in TCPOps.td docs.

One interesting observation is that when lowering tcf.matmul to
linalg.matmul, we need to both 1) create the error checks and 2)
calculate a shape transfer function to create the init tensors.
Previously, 2) was deferred to bufferizing tcp.matmul later. I'm not
sure if this is a conflation of concerns or not. For now, it's not a big
burden.

2020-11-10 18:58:28 -08:00

2 Commits (58c703010452bfbc5eabef5543058335c2e2f7b2)