Commit 9ab7527
authored
Consolidate overloads in torchlib (#2604)
The goal is to have no overloads and remove the PyTorch dispatcher.
Right now there are still the following ops that need to be addressed:
```
Registering private function: aten::as_strided
Registering private function: aten::embedding_bag
Registering private function: aten::embedding_bag.padding_idx
Registering overload for function: aten::index.Tensor
Registering overload for function: aten::_unsafe_index.Tensor
Registering overload for function: aten::index_put
```
I did a bit of cleaning up in tests and torchlib as well.
#2580
---------
Signed-off-by: Justin Chu <[email protected]>1 parent 075fc4d commit 9ab7527
File tree
7 files changed
+308
-1092
lines changed- onnxscript
- backend
- function_libs/torch_lib/ops
- requirements/ci
- tests/function_libs/torch_lib
7 files changed
+308
-1092
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
29 | 29 | | |
30 | 30 | | |
31 | 31 | | |
32 | | - | |
33 | | - | |
34 | | - | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
35 | 35 | | |
36 | 36 | | |
37 | 37 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
84 | 84 | | |
85 | 85 | | |
86 | 86 | | |
| 87 | + | |
87 | 88 | | |
88 | 89 | | |
89 | 90 | | |
| |||
0 commit comments