Skip to content

Commit

Permalink
[CI] Upgrade to linux_job_v2
Browse files Browse the repository at this point in the history 
ghstack-source-id: cf280238d4a241c02edb5d3e8ffb52c24fd228c9
Pull Request resolved: #1097
  • Loading branch information
@vmoens
vmoens committed Nov 20, 2024
1 parent 978eb6c commit f606f7b
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 8 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/docs.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ jobs:
matrix:
python_version: ["3.10"]
cuda_arch_version: ["12.1"]
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
repository: pytorch/tensordict
upload-artifact: docs
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/lint.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ concurrency:

jobs:
python-source-and-configs:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
repository: pytorch/tensordict
script: |
Expand Down Expand Up @@ -45,7 +45,7 @@ jobs:
echo '::endgroup::'
c-source:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
repository: pytorch/tensordict
script: |
Expand Down
8 changes: 4 additions & 4 deletions .github/workflows/test-linux.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ jobs:
python_version: ["3.10"]
cuda_arch_version: ["12.1"]
fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
runner: linux.g5.4xlarge.nvidia.gpu
repository: pytorch/tensordict
Expand Down Expand Up @@ -56,7 +56,7 @@ jobs:
matrix:
python_version: ["3.9", "3.10", "3.11", "3.12"]
fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
runner: linux.12xlarge
repository: pytorch/tensordict
Expand All @@ -83,7 +83,7 @@ jobs:
python_version: ["3.10"]
cuda_arch_version: ["12.1"]
fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
runner: linux.g5.4xlarge.nvidia.gpu
repository: pytorch/tensordict
Expand Down Expand Up @@ -115,7 +115,7 @@ jobs:
matrix:
python_version: ["3.9", "3.12"]
fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
runner: linux.12xlarge
repository: pytorch/tensordict
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/test-rl-gpu.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ jobs:
python_version: ["3.10"]
cuda_arch_version: ["12.1"]
fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
with:
runner: linux.g5.4xlarge.nvidia.gpu
repository: pytorch/tensordict
Expand Down

1 comment on commit f606f7b

@github-actions
Copy link

Choose a reason for hiding this comment

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

⚠️ Performance Alert ⚠️

Possible performance regression was detected for benchmark 'GPU Benchmark Results'.
Benchmark result of this commit is worse than the previous benchmark result exceeding threshold 2.

Benchmark suite Current: f606f7b Previous: 978eb6c Ratio
benchmarks/common/common_ops_test.py::test_membership_stacked_nested_last 129412.20865633577 iter/sec (stddev: 5.546844799207036e-7) 355314.86415975675 iter/sec (stddev: 3.358965780357628e-7) 2.75
benchmarks/common/common_ops_test.py::test_membership_stacked_nested_leaf_last 129020.56601124161 iter/sec (stddev: 8.076424239053053e-7) 356749.3022176497 iter/sec (stddev: 3.3547317727582734e-7) 2.77

This comment was automatically generated by workflow using github-action-benchmark.

CC: @vmoens

Please sign in to comment.