torch-mlir/build_tools/python_deploy
Ramiro Leal-Cavazos a710237437
[custom op] Generalize shape library logic to work with dtypes (#1594)
* [custom op] Generalize shape library logic to work with dtypes

This commit generalizes the shape library logic, so that dtype rules
for ops can also be expressed using the same mechanism. In other
words, each op can now have a shape function and a dtype function
specified in Python that is imported during lowering to calculate the
shapes and dtypes throught a program. For more information about how
to specify a dtype function, see the updated
`docs/adding_a_shape_and_dtype_function.md`.

For those not familiar with how the shape library works, the file
`docs/calculations_lib.md` provides an overview.
2022-12-13 08:25:41 -08:00
..
.gitignore Add oneshot release snapshot for test/ondemand (#768) 2022-04-21 02:19:12 -07:00
build_linux_packages.sh [custom op] Generalize shape library logic to work with dtypes (#1594) 2022-12-13 08:25:41 -08:00
build_macos_packages.sh Reenable LTC in out-of-tree build (for real this time) (#1205) 2022-08-19 15:25:00 -04:00
build_upload_m1_snapshot.sh build: improve robustness of cmake and shell scripts (#1018) 2022-07-06 14:39:30 -07:00
build_windows.ps1 ci: re-add powershell script for windows release builds (#1561) 2022-11-06 12:48:38 -06:00
build_windows_ci.sh ci: enable ccache on Windows (#1548) 2022-11-03 12:17:22 -05:00
install_macos_deps.sh Fix OSX nightly builds (#1032) 2022-07-10 22:17:01 -07:00