Add torch conditionals: IfElse #940
Closed
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Ch0ronomato commented Jul 17, 2024
ricardoV94 reviewed Jul 18, 2024
* Add `OpFromGraph` wrapper around `alloc_diag` * Remove depreciated `AllocDiag` `Op`, rename `AllocDiag2 -> AllocDiag` * Set `inline = False` * Add rewrite to inline all `OpFromGraph` `Op`s * Add `is_zero_offset` helper to `Eye` * Add `is_left_expand_dims` and `is_right_expand_dims` attributes to `DimShuffle` * Seed `test_local_lift_through_linalg` test
48 tasks
* Bump actions/upload-artifact from 3 to 4 Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact) from 3 to 4. - [Release notes](https://github.com/actions/upload-artifact/releases) - [Commits](actions/upload-artifact@v3...v4) --- updated-dependencies: - dependency-name: actions/upload-artifact dependency-type: direct:production update-type: version-update:semver-major ... Signed-off-by: dependabot[bot] <support@github.com> * Migrate pypi to GHA artifacts v4 * Bump download-artifact to v4 * Eliminate undefined matrix.python-version variable * Upload/download each platform separately * Use pattern arg to download-artifact --------- Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Ben Mares <services-git-throwaway1@tensorial.com>
updates: - [github.com/astral-sh/ruff-pre-commit: v0.5.2 → v0.5.4](astral-sh/ruff-pre-commit@v0.5.2...v0.5.4)
* Fixed dead wiki links * Fixed dead wiki links * Deleted old documentation at doc/sandbox.
Co-authored-by: Ben Mares <services-git-throwaway1@tensorial.com>
* Fixed dead wiki links * Fixed dead wiki links * Deleted old documentation at doc/sandbox.
Co-authored-by: Ben Mares <services-git-throwaway1@tensorial.com>
updates: - [github.com/astral-sh/ruff-pre-commit: v0.5.4 → v0.5.5](astral-sh/ruff-pre-commit@v0.5.4...v0.5.5)
Co-authored-by: Adrian Seyboldt <aseyboldt@users.noreply.github.com> Co-authored-by: Jesse Grabowski <48652735+jessegrabowski@users.noreply.github.com> Co-authored-by: Ricardo Vieira <28983449+ricardov94@users.noreply.github.com> Co-authored-by: Rob Zinkov <zaxtax@users.noreply.github.com>
…tributing guidelines.
updates: - [github.com/astral-sh/ruff-pre-commit: v0.5.5 → v0.5.6](astral-sh/ruff-pre-commit@v0.5.5...v0.5.6)
* Add building of pyodide universal wheels * precommit * Fix precommit. Readd comment. * Fix precommit2 * Minor improvement to ext_modules conditional definition * Bump Python version so that tomllib is included This way versioneer can read pyproject.toml * Add wheel package to build dependencies * Update .github/workflows/pypi.yml * Revert unnecessary * ruff --------- Co-authored-by: Ben Mares <services-git-throwaway1@tensorial.com>
fix: cast to elemwise outputs to their respective dtypes fix: Relax scipy dependency, should work in both cases style: black wrap with asarray fix: make elemwise test check against dtype in the graph fix scalar issues Update pytensor/scalar/basic.py Co-authored-by: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com> fix test add a clarifying comment to checking nan fix: bool is deprecated in numpy deps: bound scipy version improve test
Contributor Author
| hmmm - this branch got messed up - i'm gonna close the branch and make a new one. |
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
11 participants
Add this suggestion to a batch that can be applied as a single commit. This suggestion is invalid because no changes were made to the code. Suggestions cannot be applied while the pull request is closed. Suggestions cannot be applied while viewing a subset of changes. Only one suggestion per line can be applied in a batch. Add this suggestion to a batch that can be applied as a single commit. Applying suggestions on deleted lines is not supported. You must change the existing code in this line in order to create a valid suggestion. Outdated suggestions cannot be applied. This suggestion has been applied or marked resolved. Suggestions cannot be applied from pending reviews. Suggestions cannot be applied on multi-line comments. Suggestions cannot be applied while the pull request is queued to merge. Suggestion cannot be applied right now. Please check back later.
Description
Adds the IfElse node for torch, using
torch.whereRelated Issue
Checklist
Type of change