torch-mlir/python
Quinn Dawkins 85f383ce0b
Bump the shape lib to match the upstream functions currently in PyTorch (#1236)
Bumps the shape library:
 - Updates the function signature for aten.arange.start_step
 - upstream_shape_functions.mean_dim -> upstream_shape_functions.sum_mean_dim
2022-08-17 00:11:04 -04:00
..
test Propagate device data names (#1157) 2022-08-16 09:30:22 -04:00
torch_mlir Bump the shape lib to match the upstream functions currently in PyTorch (#1236) 2022-08-17 00:11:04 -04:00
torch_mlir_e2e_test Re-enable custom op support 2022-08-16 22:49:08 +05:30
CMakeLists.txt Re-enable custom op support 2022-08-16 22:49:08 +05:30
TorchMLIRModule.cpp Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00