Skip to content
This repository was archived by the owner on Nov 1, 2024. It is now read-only.

Commit 95c7dd0

Browse files
authored
Bump to v0.2.0 (#30)
1 parent b55596f commit 95c7dd0

25 files changed

+205
-46
lines changed

.github/workflows/_test_conda.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ jobs:
4242
--channel ~/conda-build\
4343
--channel pytorch-nightly\
4444
--channel conda-forge\
45-
numpy==1.21.5\
45+
numpy\
4646
pytest==7.0.1\
4747
python==${{ matrix.py }}\
4848
torchdistx

.github/workflows/_test_wheel.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ jobs:
4141
4242
source ~/venvs/test/bin/activate
4343
44-
pip install ~/wheelhouse/*.whl numpy==1.21.5\
44+
pip install ~/wheelhouse/*.whl\
4545
--requirement use-${{ matrix.build_variant }}.txt\
4646
--requirement requirements-devel.txt\
4747
--no-cache-dir

.github/workflows/nightly.yaml

Lines changed: 37 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ jobs:
3333
with:
3434
matrix: |
3535
{
36-
py: ['3.7', '3.8', '3.9'],
37-
build_variant: ['cpu', 'cu102', 'cu113'],
36+
py: ['3.7', '3.8', '3.9', '3.10'],
37+
build_variant: ['cpu', 'cu102', 'cu113', 'cu116'],
3838
sanitizer: ['nosan'],
3939
include: [
4040
{
@@ -58,8 +58,8 @@ jobs:
5858
with:
5959
matrix: |
6060
{
61-
py: ['3.7', '3.8', '3.9'],
62-
build_variant: ['cpu', 'cu102', 'cu113'],
61+
py: ['3.7', '3.8', '3.9', '3.10'],
62+
build_variant: ['cpu', 'cu102', 'cu113', 'cu116'],
6363
sanitizer: ['nosan'],
6464
include: [
6565
{
@@ -83,7 +83,7 @@ jobs:
8383
with:
8484
matrix: |
8585
{
86-
py: ['3.7', '3.8', '3.9'],
86+
py: ['3.7', '3.8', '3.9', '3.10'],
8787
build_variant: ['cpu'],
8888
sanitizer: ['nosan'],
8989
include: [
@@ -107,7 +107,7 @@ jobs:
107107
with:
108108
matrix: |
109109
{
110-
py: ['3.7', '3.8', '3.9'],
110+
py: ['3.7', '3.8', '3.9', '3.10'],
111111
build_variant: ['cu102'],
112112
sanitizer: ['nosan']
113113
}
@@ -119,19 +119,31 @@ jobs:
119119
with:
120120
matrix: |
121121
{
122-
py: ['3.7', '3.8', '3.9'],
122+
py: ['3.7', '3.8', '3.9', '3.10'],
123123
build_variant: ['cu113'],
124124
sanitizer: ['nosan']
125125
}
126126
127+
test_wheel_cu116:
128+
name: Test (CUDA 11.6)
129+
needs: test_wheel_cu113
130+
uses: ./.github/workflows/_test_wheel.yaml
131+
with:
132+
matrix: |
133+
{
134+
py: ['3.7', '3.8', '3.9', '3.10'],
135+
build_variant: ['cu116'],
136+
sanitizer: ['nosan']
137+
}
138+
127139
test_conda_cpu:
128140
name: Test (CPU)
129141
needs: build_conda
130142
uses: ./.github/workflows/_test_conda.yaml
131143
with:
132144
matrix: |
133145
{
134-
py: ['3.7', '3.8', '3.9'],
146+
py: ['3.7', '3.8', '3.9', '3.10'],
135147
build_variant: ['cpu'],
136148
sanitizer: ['nosan'],
137149
include: [
@@ -155,7 +167,7 @@ jobs:
155167
with:
156168
matrix: |
157169
{
158-
py: ['3.7', '3.8', '3.9'],
170+
py: ['3.7', '3.8', '3.9', '3.10'],
159171
build_variant: ['cu102'],
160172
sanitizer: ['nosan']
161173
}
@@ -167,21 +179,33 @@ jobs:
167179
with:
168180
matrix: |
169181
{
170-
py: ['3.7', '3.8', '3.9'],
182+
py: ['3.7', '3.8', '3.9', '3.10'],
171183
build_variant: ['cu113'],
172184
sanitizer: ['nosan']
173185
}
174186
187+
test_conda_cu116:
188+
name: Test (CUDA 11.6)
189+
needs: test_conda_cu113
190+
uses: ./.github/workflows/_test_conda.yaml
191+
with:
192+
matrix: |
193+
{
194+
py: ['3.7', '3.8', '3.9', '3.10'],
195+
build_variant: ['cu116'],
196+
sanitizer: ['nosan']
197+
}
198+
175199
deploy:
176200
name: Deploy
177201
if: github.event_name == 'schedule' || github.event.inputs.deploy == 'true'
178-
needs: [build_doc, test_wheel_cu113, test_conda_cu113]
202+
needs: [build_doc, test_wheel_cu116, test_conda_cu116]
179203
uses: ./.github/workflows/_deploy.yaml
180204
with:
181205
matrix: |
182206
{
183-
py: ['3.7', '3.8', '3.9'],
184-
build_variant: ['cpu', 'cu102', 'cu113'],
207+
py: ['3.7', '3.8', '3.9', '3.10'],
208+
build_variant: ['cpu', 'cu102', 'cu113', 'cu116'],
185209
}
186210
s3_wheel_path: pytorch/whl/nightly
187211
doc_folder_override: nightly

.github/workflows/push.yaml

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ jobs:
3434
matrix: |
3535
{
3636
py: ['3.7'],
37-
build_variant: ['cpu', 'cu102', 'cu113'],
37+
build_variant: ['cpu', 'cu102', 'cu113', 'cu116'],
3838
sanitizer: ['nosan'],
3939
include: [
4040
{
@@ -93,3 +93,19 @@ jobs:
9393
},
9494
]
9595
}
96+
97+
test_wheel_cu116:
98+
name: Test (CUDA 11.6)
99+
needs: test_wheel_cu113
100+
uses: ./.github/workflows/_test_wheel.yaml
101+
with:
102+
matrix: |
103+
{
104+
include: [
105+
{
106+
py: '3.7',
107+
build_variant: 'cu116',
108+
sanitizer: 'nosan'
109+
},
110+
]
111+
}

.github/workflows/release.yaml

Lines changed: 37 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ jobs:
3030
with:
3131
matrix: |
3232
{
33-
py: ['3.7', '3.8', '3.9'],
34-
build_variant: ['cpu', 'cu102', 'cu113'],
33+
py: ['3.7', '3.8', '3.9', '3.10'],
34+
build_variant: ['cpu', 'cu102', 'cu113', 'cu116'],
3535
sanitizer: ['nosan'],
3636
include: [
3737
{
@@ -54,8 +54,8 @@ jobs:
5454
with:
5555
matrix: |
5656
{
57-
py: ['3.7', '3.8', '3.9'],
58-
build_variant: ['cpu', 'cu102', 'cu113'],
57+
py: ['3.7', '3.8', '3.9', '3.10'],
58+
build_variant: ['cpu', 'cu102', 'cu113', 'cu116'],
5959
sanitizer: ['nosan'],
6060
include: [
6161
{
@@ -78,7 +78,7 @@ jobs:
7878
with:
7979
matrix: |
8080
{
81-
py: ['3.7', '3.8', '3.9'],
81+
py: ['3.7', '3.8', '3.9', '3.10'],
8282
build_variant: ['cpu'],
8383
sanitizer: ['nosan'],
8484
include: [
@@ -102,7 +102,7 @@ jobs:
102102
with:
103103
matrix: |
104104
{
105-
py: ['3.7', '3.8', '3.9'],
105+
py: ['3.7', '3.8', '3.9', '3.10'],
106106
build_variant: ['cu102'],
107107
sanitizer: ['nosan']
108108
}
@@ -114,19 +114,31 @@ jobs:
114114
with:
115115
matrix: |
116116
{
117-
py: ['3.7', '3.8', '3.9'],
117+
py: ['3.7', '3.8', '3.9', '3.10'],
118118
build_variant: ['cu113'],
119119
sanitizer: ['nosan']
120120
}
121121
122+
test_wheel_cu116:
123+
name: Test (CUDA 11.6)
124+
needs: test_wheel_cu113
125+
uses: ./.github/workflows/_test_wheel.yaml
126+
with:
127+
matrix: |
128+
{
129+
py: ['3.7', '3.8', '3.9', '3.10'],
130+
build_variant: ['cu116'],
131+
sanitizer: ['nosan']
132+
}
133+
122134
test_conda_cpu:
123135
name: Test (CPU)
124136
needs: build_conda
125137
uses: ./.github/workflows/_test_conda.yaml
126138
with:
127139
matrix: |
128140
{
129-
py: ['3.7', '3.8', '3.9'],
141+
py: ['3.7', '3.8', '3.9', '3.10'],
130142
build_variant: ['cpu'],
131143
sanitizer: ['nosan'],
132144
include: [
@@ -150,7 +162,7 @@ jobs:
150162
with:
151163
matrix: |
152164
{
153-
py: ['3.7', '3.8', '3.9'],
165+
py: ['3.7', '3.8', '3.9', '3.10'],
154166
build_variant: ['cu102'],
155167
sanitizer: ['nosan']
156168
}
@@ -162,21 +174,33 @@ jobs:
162174
with:
163175
matrix: |
164176
{
165-
py: ['3.7', '3.8', '3.9'],
177+
py: ['3.7', '3.8', '3.9', '3.10'],
166178
build_variant: ['cu113'],
167179
sanitizer: ['nosan']
168180
}
169181
182+
test_conda_cu116:
183+
name: Test (CUDA 11.6)
184+
needs: test_conda_cu113
185+
uses: ./.github/workflows/_test_conda.yaml
186+
with:
187+
matrix: |
188+
{
189+
py: ['3.7', '3.8', '3.9', '3.10'],
190+
build_variant: ['cu116'],
191+
sanitizer: ['nosan']
192+
}
193+
170194
deploy:
171195
name: Deploy
172196
if: github.event.inputs.deploy == 'true'
173-
needs: [build_doc, test_wheel_cu113, test_conda_cu113]
197+
needs: [build_doc, test_wheel_cu116, test_conda_cu116]
174198
uses: ./.github/workflows/_deploy.yaml
175199
with:
176200
matrix: |
177201
{
178-
py: ['3.7', '3.8', '3.9'],
179-
build_variant: ['cpu', 'cu102', 'cu113'],
202+
py: ['3.7', '3.8', '3.9', '3.10'],
203+
build_variant: ['cpu', 'cu102', 'cu113', 'cu116'],
180204
}
181205
s3_wheel_path: pytorch/whl
182206
secrets:

CHANGELOG.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,10 @@ All notable changes to this project will be documented in this file.
44
The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/),
55
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
66

7-
## [0.2.0] - 2022-mm-dd
8-
TBD
7+
## [0.2.0] - 2022-06-23
8+
- Moves to PyTorch v1.12
9+
- Adds support for Python 3.10
10+
- Addresses a minor bug in Fake tensor caused by the API changes in PyTorch
911

1012
## [0.1.0] - 2022-04-14
1113
### Added

README.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,11 @@ As of today the following features are available in torchdistX:
2020
## Dependencies
2121
torchdistX versions corresponding to each PyTorch release:
2222

23-
| `torch` | `torchdistx` | `python` |
24-
| ------------ | ------------ | ---------------- |
25-
| `main` | `main` | `>=3.7`, `<=3.9` |
26-
| `1.11.0` | `0.1.0` | `>=3.7`, `<=3.9` |
23+
| `torch` | `torchdistx` | `python` |
24+
| ------------ | ------------ | ----------------- |
25+
| `main` | `main` | `>=3.7`, `<=3.10` |
26+
| `1.12.0` | `0.2.0` | `>=3.7`, `<=3.10` |
27+
| `1.11.0` | `0.1.0` | `>=3.7`, `<=3.9` |
2728

2829
## Installation
2930
As of today only Linux and macOS operating systems are supported. Please note

VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0.2.0-dev
1+
0.2.0

docker/ci-conda/Dockerfile.cu116

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
# Copyright (c) Meta Platforms, Inc. and affiliates.
2+
# All rights reserved.
3+
#
4+
# This source code is licensed under the BSD-style license found in the
5+
# LICENSE file in the root directory of this source tree.
6+
7+
FROM ghcr.io/pytorch/torchdistx-ci-conda:1-cpu
8+
9+
COPY install-cuda-11.6 /root/
10+
11+
RUN /root/install-cuda-11.6

docker/ci-conda/install-cuda-11.6

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
#!/usr/bin/env bash
2+
3+
# Copyright (c) Meta Platforms, Inc. and affiliates.
4+
# All rights reserved.
5+
#
6+
# This source code is licensed under the BSD-style license found in the
7+
# LICENSE file in the root directory of this source tree.
8+
9+
set -o errexit
10+
11+
curl --location --fail --output cuda.run\
12+
https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
13+
14+
sh cuda.run --silent --toolkit --override --no-man-page
15+
16+
rm cuda.run

docker/ci-wheel/Dockerfile.cu116

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
# Copyright (c) Meta Platforms, Inc. and affiliates.
2+
# All rights reserved.
3+
#
4+
# This source code is licensed under the BSD-style license found in the
5+
# LICENSE file in the root directory of this source tree.
6+
7+
FROM ghcr.io/pytorch/torchdistx-ci-wheel:1-cpu
8+
9+
# CUDA 11.6 requires GCC 11.x.
10+
ENV PATH=/usr/local/cuda-11.6/bin:/opt/rh/devtoolset-11/root/usr/bin:$PATH
11+
12+
ENV LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:/opt/rh/devtoolset-11/root/usr/lib64:$LD_LIBRARY_PATH
13+
14+
COPY install-devtoolset-11 install-cuda-11.6 install-cudnn-8.3.2 /root/
15+
16+
RUN /root/install-devtoolset-11
17+
RUN /root/install-cuda-11.6
18+
RUN /root/install-cudnn-8.3.2

docker/ci-wheel/install-cuda-11.6

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
#!/usr/bin/env bash
2+
3+
# Copyright (c) Meta Platforms, Inc. and affiliates.
4+
# All rights reserved.
5+
#
6+
# This source code is licensed under the BSD-style license found in the
7+
# LICENSE file in the root directory of this source tree.
8+
9+
set -o errexit
10+
11+
curl --location --fail --output cuda.run\
12+
https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
13+
14+
sh cuda.run --silent --toolkit --override --no-man-page
15+
16+
rm cuda.run

0 commit comments

Comments
 (0)