torch-mlir/lib/Dialect/Torch
jinchen62 1682b540bf
Prototype passes for lowering quantized group matmul (#2402)
* Support brevitas custom op (#2320)

* f16 change for brevitas

* Adapt the change of brevitas quant custom op name

* Add unit tests

* Make brevitas conversions isolated

* Address the comments

---------

Co-authored-by: dan <danimal197@gmail.com>
2023-08-29 21:25:45 -07:00
..
IR Prototype passes for lowering quantized group matmul (#2402) 2023-08-29 21:25:45 -07:00
Transforms [MLIR][TORCH] add E2E support for aten.min op (#2422) 2023-08-29 12:12:41 -05:00
Utils [Torch-Dialect] emit aten.narrow.Tensor op and decompose it to aten.narrow op (#2297) 2023-07-20 16:46:44 +08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00