Skip to content

Move torch.compile to HPUModelRunnerBase#966

Merged
michalkuligowski merged 2 commits intohabana_mainfrom
dev/anko/broader_compilation
Apr 1, 2025
Merged

Move torch.compile to HPUModelRunnerBase#966
michalkuligowski merged 2 commits intohabana_mainfrom
dev/anko/broader_compilation

Conversation

@anko-intel
Copy link
Copy Markdown

@anko-intel anko-intel commented Mar 26, 2025

To narrow the gap between lazy and compile mode, PyTorch compilations is moved to the place where HPUGraph are collected.
This way code from HpuModelAdapter can also be compiled.
For regional compilation this change add compilations for some of HpuModelAdapter methods.

+ compile some of HpuModelAdapter methods in regional compilation
Copy link
Copy Markdown

@jczaja jczaja left a comment

Choose a reason for hiding this comment

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

LGTM

@anko-intel
Copy link
Copy Markdown
Author

/run-gaudi-tests

1 similar comment
@afierka-intel
Copy link
Copy Markdown

/run-gaudi-tests

@michalkuligowski
Copy link
Copy Markdown

/run-gaudi-tests

@michalkuligowski michalkuligowski merged commit 3f724ff into habana_main Apr 1, 2025
39 checks passed
@michalkuligowski michalkuligowski deleted the dev/anko/broader_compilation branch April 1, 2025 07:55
adobrzyn pushed a commit that referenced this pull request May 13, 2025
I checked following PRs and moved t.compile optimizations from V0 to V1
#966
#994
#1078
#1140
#1143
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.

4 participants