mirror of https://github.com/llvm/torch-mlir
234fc7fe0c
Prior to this patch, the torch dialect included `AtenTriuOp` for computing the upper triangular part of the input matrix, but there was no code for lowering the op to the linalg dialect. This patch adds code to generate a `linalg.generic` operation that compares indices (computed using `linalg.index`) to choose between zero or the original value (using `arith.select`). The lowering fails if the number of dimensions are less than two. This patch also adds a few end-to-end tests. |
||
|---|---|---|
| .. | ||
| _torch_mlir_custom_op_example | ||
| dialects | ||
| eager_mode | ||
| __init__.py | ||
| compiler_utils.py | ||