torch-mlir

Commit Graph

Author	SHA1	Message	Date
Vivek Khandelwal	5dbace239b	Extend tm_tensor.scatter op semantic to carry unique_indices attribute There are cases where the op may update the same indices multiple times. In this context, we can not parallelize updates. Instead, we have to execute them sequentially. Adding a boolean attribute to control the behavior. Also adding test cases for invalid IR.	2022-02-22 22:02:10 +05:30
Nirvedh	f8cb32faf0	LLVM bump Major changes: opTrait changed to Trait, selectOp moved to arith dialect assertOp moved to cf dialect	2022-02-16 15:28:13 -05:00
Yi Zhang	869daf3c22	Add TMTensor dialect to torch-mlir This is intended to explore support for non-structured ops that can't be modeled by Linalg dialect. `tm_tensor.scan` and `tm_tensor.scatter` are added as the first such ops. The dialect should aim to be upstreamed in the future.	2022-02-15 16:45:38 -05:00

Author

SHA1

Message

Date

Vivek Khandelwal

5dbace239b

Extend tm_tensor.scatter op semantic to carry unique_indices attribute

There are cases where the op may update the same indices multiple
times. In this context, we can not parallelize updates. Instead,
we have to execute them sequentially. Adding a boolean attribute
to control the behavior.

Also adding test cases for invalid IR.

2022-02-22 22:02:10 +05:30

Nirvedh

f8cb32faf0

LLVM bump

Major changes: opTrait changed to Trait, selectOp moved to arith dialect
assertOp moved to cf dialect

2022-02-16 15:28:13 -05:00

Yi Zhang

869daf3c22

Add TMTensor dialect to torch-mlir

This is intended to explore support for non-structured ops that can't
be modeled by Linalg dialect. `tm_tensor.scan` and `tm_tensor.scatter`
are added as the first such ops. The dialect should aim to be
upstreamed in the future.

2022-02-15 16:45:38 -05:00

3 Commits (819f29316f961cf77a754ea2567f66a8a35cc493)