mirror of https://github.com/llvm/torch-mlir
2e194e13d6
This PR fixes the bugs for `Torch::AtenOneHotOp` by: 1) Using `Torch::kUnknownSize` as the default value for `numClasses` in the pattern matching stage in `DecomposeAtenOneHotOp` 2) Adding `AtenIntScalarOp` to the patterns in `TorchToArith` 3) Handling both `int` and `float` types for `off` and `on` values in `TorchOnnxToTorch` conversion It also includes: 1) A new test in `TorchToArith/basic.mlir`, for `torch.aten.Int.Scalar`, and 2) A new test in `decompose-complex-ops.mlir`, for `torch.aten.one_hot` **Dependencies** This PR is dependent on #3334. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| DefaultDomainAtoF.cpp | ||
| DefaultDomainGtoP.cpp | ||
| DefaultDomainQtoZ.cpp | ||
| OnnxLstmExpander.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| Patterns.cpp | ||
| TorchOnnxToTorch.cpp | ||
| Utils.cpp | ||