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.

Sign up for GitHub

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

Jump to bottom

[Torch] Bump PyTorch Version to 2.6.0 #3253

Merged
merged 48 commits into from
Feb 24, 2025
Merged

[Torch] Bump PyTorch Version to 2.6.0 #3253

merged 48 commits into from
Feb 24, 2025
+244 −181

Conversation

anzr299
Copy link
Collaborator

@anzr299 anzr299 commented Feb 4, 2025 •
edited
Loading

Changes

  1. Bump Torch version up to 2.6.0 and torchvision to 0.21.0
  2. use weights_only=False for all model loading in pytorch
  3. modify multi_head_attention_forward code in pt2 function hook
  4. xfail torch fx backend tests in conformance test issue-162009
  5. do not export pytorch backend models before ov.convert in conformance test.
  6. transformers version update from 4.38.2 -> 4.48.3, optimum 1.23.3 -> 1.24.0, optimum-intel 1.21.0 -> 1.22.0, accelerate 0.28.0 -> 1.1.0
  7. make attn_implementation=eager to ensure old transformers SDPA attention is used instead of the new torch SDPA API
  8. add xfail for swin model in test_model_saving due to issue-162383
Category Job Status Job Number Notes
Nightly install_pt_cpu Pass 637  
Nightly install_pt_gpu Pass 638  
Nightly torch_nightly Pass 463  
Nightly e2e_pytorch_full Pass 962  
Nightly/Windows install_torch_cpu Pass 526  
Nightly/Windows install_torch_gpu - - No Build info
Manual post_training_quantization Pass 623  
Manual post_training_weights_compression Pass 326  
GH WC WC test on GHA Pass 33
Example Example Pass 238  

@anzr299
init
Loading
Loading status checks…
3c8933e
@anzr299 anzr299 requested a review from a team as a code owner February 4, 2025 17:35
@github-actions github-actions bot added documentation Improvements or additions to documentation NNCF PT Pull requests that updates NNCF PyTorch labels Feb 4, 2025
@anzr299
fix weights only issue
Loading
Loading status checks…
56c2766
@anzr299
update graphs
Loading
Loading status checks…
d4e3942
@anzr299
update graphs
Loading
Loading status checks…
816016e
@anzr299
update model loading instances
Loading
Loading status checks…
bfa8f62
@github-actions github-actions bot added the NNCF PTQ Pull requests that updates NNCF PTQ label Feb 5, 2025
anzr299 added 15 commits February 13, 2025 11:16
@anzr299
update transformers requirement
f6c338e 
pre-commit fix

remove debug statements

accelerate update
@anzr299
Merge branch 'openvinotoolkit:develop' into an/torch260
2d8b9b8
@anzr299
load_weight and torch 2 fixes
1b31e79
@anzr299
big fix: use self attention instead of SDPA while loading hf models
b14e4aa
@anzr299
part of big fix
e0fb153
@anzr299
missed out model hf fix
3fbea82
@anzr299
Merge branch 'develop' into an/torch260
00255ea
@anzr299
pre-commit fix
719b4ba
@anzr299
Merge branch 'an/torch260' of https://github.com/anzr299/nncf into an…
27b93fe 
…/torch260
@anzr299
fix comments
221dc0f
@anzr299
update reference graphs
8e742de
@anzr299
change torch.export for swin model to torch.export.export_for_inference
a230b47
@anzr299
update optimum intel version
d230ec5 
update optimum-intel version

update requirements

change to export for inference

fix pt issue

pre-commit fix
@anzr299
Merge branch 'develop' into an/torch260
ee50c4f
@anzr299
revert old pt fix
Loading
Loading status checks…
527d301 
xfail for known issue

pre-commit fix
@anzr299
Merge branch 'openvinotoolkit:develop' into an/torch260

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode
Loading
Loading status checks…
e99e4c2
anzr299 and others added 8 commits February 19, 2025 13:39
@anzr299
modify issue references for appropriation
Loading
Loading status checks…
f9d8c56
@anzr299
update requirements for conformance test to include accelerate
Loading
Loading status checks…
59e1b00
@anzr299
add proper xfail for FX backend
d036282
@anzr299
fix
5e3476d
@anzr299
do not export for OV backend in torhcvision pipeline
Loading
Loading status checks…
844ff75
@anzr299
pre commit fix
Loading
Loading status checks…
a4f1a0f
@anzr299
update xfail condition
Loading
Loading status checks…
1873d94
@AlexanderDokuchaev
remove debug code

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode
Loading
Loading status checks…
a92fad0
@github-actions github-actions bot added the NNCF OpenVINO Pull requests that updates NNCF OpenVINO label Feb 20, 2025
@alexsu52 alexsu52 self-requested a review February 20, 2025 14:24
alexsu52
alexsu52 reviewed Feb 21, 2025
View reviewed changes
Copy link
Contributor

@alexsu52 alexsu52 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please update actual status for post_training_quantization build in the description.

@anzr299
Copy link
Collaborator Author

anzr299 commented Feb 21, 2025

Please update actual status for post_training_quantization build in the description.

Done

@anzr299
Merge branch 'develop' into an/torch260

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode
Loading
Loading status checks…
748871c
@github-actions github-actions bot removed the NNCF OpenVINO Pull requests that updates NNCF OpenVINO label Feb 21, 2025
@alexsu52 alexsu52 merged commit 8d33e42 into openvinotoolkit:develop Feb 24, 2025
17 checks passed
@nikita-malininn nikita-malininn mentioned this pull request Feb 24, 2025
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@alexsu52 alexsu52

@AlexanderDokuchaev AlexanderDokuchaev

Assignees

@AlexanderDokuchaev AlexanderDokuchaev

Labels
documentation Improvements or additions to documentation experimental NNCF PT Pull requests that updates NNCF PyTorch NNCF PTQ Pull requests that updates NNCF PTQ
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
3 participants
@anzr299 @alexsu52 @AlexanderDokuchaev