Skip to content

Commit f379967

Browse files
authored
cherry-pick: Docker fixes release/2.2 (#2628)
1 parent 18a462f commit f379967

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

.github/workflows/docker_builder.yml

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
name: 'Torch-TensorRT Docker Build'
22

3-
# Apply workflow only to main branch
3+
# Apply workflow only to main, nightly, release branches
44
on:
55
push:
66
branches:
77
- main
88
- nightly
9-
- release/2.1
9+
- release/2.2
1010

1111
# If pushes to main are made in rapid succession,
1212
# cancel existing docker builds and use newer commits
@@ -16,7 +16,10 @@ concurrency:
1616

1717
jobs:
1818
build:
19-
runs-on: linux.2xlarge
19+
permissions:
20+
packages: write
21+
22+
runs-on: linux.4xlarge.nvidia.gpu
2023

2124
# Define key environment variables
2225
# Container name is of the form torch_tensorrt:<branch_name>

0 commit comments

Comments
 (0)