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

Add possibility to run training benchmarks on XPU device #7925

Merged

Conversation

DamianSzwichtenberg
Copy link
Member

Add possibility to run training benchmarks on XPU device.
Exemplary CMD:
python training_benchmark.py --device xpu --datasets Reddit --models gcn --num-layers 2 --num-hidden-channels 128 --batch-sizes 8192

Copy link
Member

@akihironitta akihironitta left a comment

Choose a reason for hiding this comment

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

LGTM! We could potentially consider adding xpu to benchmark PyG on via #7795 if we have access to an instance with a xpu device.

@DamianSzwichtenberg DamianSzwichtenberg merged commit 32a76cb into pyg-team:master Aug 28, 2023
erfanloghmani pushed a commit to erfanloghmani/pytorch_geometric that referenced this pull request Aug 31, 2023
Exemplary CMD:
`python training_benchmark.py --device xpu --datasets Reddit --models
gcn --num-layers 2 --num-hidden-channels 128 --batch-sizes 8192`

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
JakubPietrakIntel pushed a commit that referenced this pull request Sep 27, 2023
Exemplary CMD:
`python training_benchmark.py --device xpu --datasets Reddit --models
gcn --num-layers 2 --num-hidden-channels 128 --batch-sizes 8192`

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants