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

disable some models for torchbench #6352

Merged
merged 16 commits into from
Jan 24, 2024
Merged

disable some models for torchbench #6352

merged 16 commits into from
Jan 24, 2024

Conversation

zpcore
Copy link
Collaborator

@zpcore zpcore commented Jan 22, 2024

  1. Disable opacus_cifar10 for TPU due to stackdump issue.
  2. Disable cm3leon_generate, hf_T5_generate for GPU due to no install.py in torchbench.

benchmarks/torchbench_model.py Outdated Show resolved Hide resolved
benchmarks/torchbench_model.py Outdated Show resolved Hide resolved
benchmarks/experiment_runner.py Outdated Show resolved Hide resolved
Copy link
Collaborator

@ysiraichi ysiraichi left a comment

Choose a reason for hiding this comment

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

Just one more comment, but LGTM otherwise.

benchmarks/torchbench_model.py Show resolved Hide resolved
@zpcore zpcore merged commit a9005e3 into master Jan 24, 2024
18 checks passed
@zpcore zpcore deleted the piz/skipmodel branch January 24, 2024 21:49
bhavya01 pushed a commit that referenced this pull request Apr 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants