Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support PyTorch 2.0.1 on branch main #1155

Merged
merged 34 commits into from
May 23, 2023
Merged

Conversation

ClaudiaComito
Copy link
Contributor

Description

same as #1134, changed branch name to trigger CI with suitable nvidia image

Changes proposed:

Type of change

Memory requirements

Performance

Due Diligence

  • All split configurations tested
  • Multiple dtypes tested in relevant functions
  • Documentation updated (if needed)
  • Title of PR is suitable for corresponding CHANGELOG entry

Does this change modify the behaviour of other functions? If so, which?

no

ClaudiaComito and others added 30 commits November 1, 2022 14:46
* add pr workflow

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update README.md

* Update .gitlab-ci.yml

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
… torch.Tensor.item() in torch 2.0 (#1149)

* Support latest PyTorch release

* Fixed bug in optimizer due to change in default behaviour of zero_grad() function in torch 2.0

* Fixed bug in due to change in behaviour of torch.Tensor.item() in torch 2.0

* Fixed version in setup.py

* Modified tests to work for torch 1.0 also

---------

Co-authored-by: ClaudiaComito <ClaudiaComito@users.noreply.github.com>
@ClaudiaComito ClaudiaComito requested a review from mtar May 19, 2023 13:21
@ghost
Copy link

ghost commented May 19, 2023

👇 Click on the image for a new way to code review

Review these changes using an interactive CodeSee Map

Legend

CodeSee Map legend

@github-actions
Copy link
Contributor

Thank you for the PR!

@github-actions
Copy link
Contributor

Thank you for the PR!

mtar
mtar previously approved these changes May 23, 2023
@ClaudiaComito ClaudiaComito merged commit 468999e into main May 23, 2023
@ClaudiaComito ClaudiaComito deleted the support/support-pytorch-2.0.1 branch May 23, 2023 17:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants