Skip to content

Commit

Permalink
Update manylinux container tags (#156)
Browse files Browse the repository at this point in the history
Bumping up tlcpack/package-* versions to latest
docker images that include support for Arm(R) Ethos(TM)-N
NPU driver 22.11. This should fix tlcpack nightly and pypi CI.
  • Loading branch information
ashutosh-arm authored Jan 23, 2023
1 parent b4ba2bb commit 29bddb1
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 12 deletions.
10 changes: 5 additions & 5 deletions .github/workflows/conda_gpu_nigthly.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,16 +17,16 @@ jobs:
pkg: ['tlcpack', 'tlcpack-nightly']
config:
- cuda: '10.2'
image: 'tlcpack/package-cu102:7bfdf7e'
image: 'tlcpack/package-cu102:2fddbe0'
# Conda forge is not shipping a cudnn package with support for CUDA 11.0
# - cuda: '11.0'
# image: 'tlcpack/package-cu110:7bfdf7e'
# image: 'tlcpack/package-cu110:2fddbe0'
- cuda: '11.1'
image: 'tlcpack/package-cu111:7bfdf7e'
image: 'tlcpack/package-cu111:2fddbe0'
- cuda: '11.3'
image: 'tlcpack/package-cu113:7bfdf7e'
image: 'tlcpack/package-cu113:2fddbe0'
- cuda: '11.6'
image: 'tlcpack/package-cu116:7bfdf7e'
image: 'tlcpack/package-cu116:2fddbe0'

runs-on: ubuntu-latest

Expand Down
12 changes: 6 additions & 6 deletions .github/workflows/wheel_manylinux_nightly.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -19,19 +19,19 @@ jobs:
# matrix of build configs
config:
- cuda: 'none'
image: 'tlcpack/package-cpu:7bfdf7e'
image: 'tlcpack/package-cpu:2fddbe0'
- cuda: '10.2'
image: 'tlcpack/package-cu102:7bfdf7e'
image: 'tlcpack/package-cu102:2fddbe0'
# CUDA 11.0 is not supported for Conda build,
# Let's drop support here as well.
# - cuda: '11.0'
# image: 'tlcpack/package-cu110:7bfdf7e'
# image: 'tlcpack/package-cu110:2fddbe0'
- cuda: '11.1'
image: 'tlcpack/package-cu111:7bfdf7e'
image: 'tlcpack/package-cu111:2fddbe0'
- cuda: '11.3'
image: 'tlcpack/package-cu113:7bfdf7e'
image: 'tlcpack/package-cu113:2fddbe0'
- cuda: '11.6'
image: 'tlcpack/package-cu116:7bfdf7e'
image: 'tlcpack/package-cu116:2fddbe0'

runs-on: ubuntu-latest

Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/wheel_manylinux_pypi.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ jobs:
# matrix of build configs
config:
- cuda: 'none'
image: 'tlcpack/package-cpu:7bfdf7e'
image: 'tlcpack/package-cpu:2fddbe0'
package_name: 'apache-tvm'

runs-on: ubuntu-latest
Expand Down

0 comments on commit 29bddb1

Please sign in to comment.