torch-mlir/lib/Dialect/Torch/IR
Sean Silva c17c0a6ba2 Fix for 0-size dim inferred incorrectly.
The issue was in the canonicalizer for torch.aten.ge.int -- in cases
where the operands were swapped, it would miscompile. This issue is
fixed and folding support generalized to `torch.aten.size.int < 0` as
well.

Fixes #716
2022-03-30 16:36:15 -07:00
..
CMakeLists.txt Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
TorchDialect.cpp Elide `!torch.` prefix in nested dialect types. 2022-03-15 17:24:08 -07:00
TorchOps.cpp Fix for 0-size dim inferred incorrectly. 2022-03-30 16:36:15 -07:00
TorchOpsODSGenerated.cpp Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
TorchTypes.cpp Add minimal support for Union types. 2022-03-29 17:45:48 -07:00
UtilsForODSGenerated.cpp Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
UtilsForODSGenerated.h Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00