Make 1st worker gpu.nvidia.medium resource class #2227
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
torch_compile_tutorial_.py
requires a gpu with capability > 7.1https://app.circleci.com/pipelines/github/pytorch/tutorials/7439/workflows/fffc8c6e-7f15-4b5b-b843-36d3e7969395/jobs/145263
gpu.nvidia.small has a Tesla P4, which has a compute capability of 6.1, so the PR switches the file to run on gpu.nvidia.medium, which has a Test T4 and a compute capability of 7.5 (https://developer.nvidia.com/cuda-gpus#compute)
This tutorial is actually on the NOT_RUN list, so I'm not actually sure how to test
Should help #2224 pass CI