Skip to content

Commit

Permalink
Update v2.0-pre (#942)
Browse files Browse the repository at this point in the history
* Update doc URL. (#821)

* Support indexing 2-axes RaggedTensor, Support slicing for RaggedTensor (#825)

* Support index 2-axes RaggedTensor, Support slicing for RaggedTensor

* Fix compiling errors

* Fix unit test

* Change RaggedTensor.data to RaggedTensor.values

* Fix style

* Add docs

* Run nightly-cpu when pushing code to nightly-cpu branch

* Prune with max_arcs in IntersectDense (#820)

* Add checking for array constructor

* Prune with max arcs

* Minor fix

* Fix typo

* Fix review comments

* Fix typo

* Release v1.8

* Create a ragged tensor from a regular tensor. (#827)

* Create a ragged tensor from a regular tensor.

* Add tests for creating ragged tensors from regular tensors.

* Add more tests.

* Print ragged tensors in a way like what PyTorch is doing.

* Fix test cases.

* Trigger GitHub actions manually. (#829)

* Run GitHub actions on merging. (#830)

* Support printing ragged tensors in a more compact way. (#831)

* Support printing ragged tensors in a more compact way.

* Disable support for torch 1.3.1

* Fix test failures.

* Add levenshtein alignment (#828)

* Add levenshtein graph

* Contruct k2.RaggedTensor in python part

* Fix review comments, return aux_labels in ctc_graph

* Fix tests

* Fix bug of accessing symbols

* Fix bug of accessing symbols

* Change argument name, add levenshtein_distance interface

* Fix test error, add tests for levenshtein_distance

* Fix review comments and add unit test for c++ side

* update the interface of levenshtein alignment

* Fix review comments

* Release v1.9

* Support a[b[i]] where both a and b are ragged tensors. (#833)

* Display import error solution message on MacOS (#837)

* Fix installation doc. (#841)

* Fix installation doc.

Remove Windows support. Will fix it later.

* Fix style issues.

* fix typos in the install instructions (#844)

* make cmake adhere to the modernized way of finding packages outside default dirs (#845)

* import torch first in the smoke tests to preven SEGFAULT (#846)

* Add doc about how to install a CPU version of k2. (#850)

* Add doc about how to install a CPU version of k2.

* Remove property setter of Fsa.labels

* Update Ubuntu version in GitHub CI since 16.04 reaches end-of-life.

* Support PyTorch 1.10. (#851)

* Fix test cases for k2.union() (#853)

* Fix out-of-boundary access (read). (#859)

* Update all the example codes in the docs (#861)

* Update all the example codes in the docs

I have run all the modified codes with  the newest version k2.

* do some changes

* Fix compilation errors with CUB 1.15. (#865)

* Update README. (#873)

* Update README.

* Fix typos.

* Fix ctc graph (make aux_labels of final arcs -1) (#877)

* Fix LICENSE location to k2 folder (#880)

* Release v1.11. (#881)

It contains bugfixes.

* Update documentation for hash.h (#887)

* Update documentation for hash.h

* Typo fix

* Wrap MonotonicLowerBound (#883)

* Wrap MonotonicLowerBound

* Add unit tests

* Support int64; update documents

* Remove extra commas after 'TOPSORTED' properity and fix RaggedTensor constructer parameter 'byte_offset' out-of-range bug. (#892)

Co-authored-by: gzchenduisheng <gzchenduisheng@corp.netease.com>

* Fix small typos (#896)

* Fix k2.ragged.create_ragged_shape2 (#901)

Before the fix, we have to specify both `row_splits` and `row_ids`
while calling `k2.create_ragged_shape2` even if one of them is `None`.

After this fix, we only need to specify one of them.

* Add rnnt loss (#891)

* Add cpp code of mutual information

* mutual information working

* Add rnnt loss

* Add pruned rnnt loss

* Minor Fixes

* Minor fixes & fix code style

* Fix cpp style

* Fix code style

* Fix s_begin values in padding positions

* Fix bugs related to boundary; Fix s_begin padding value; Add more tests

* Minor fixes

* Fix comments

* Add boundary to pruned loss tests

* Use more efficient way to fix boundaries (#906)

* Release v1.12 (#907)

* Change the sign of the rnnt_loss and add reduction argument (#911)

* Add right boundary constrains for s_begin

* Minor fixes to the interface of rnnt_loss to make it return positive value

* Fix comments

* Release a new version

* Minor fixes

* Minor fixes to the docs

* Fix building doc. (#908)

* Fix building doc.

* Minor fixes.

* Minor fixes.

* Fix building doc (#912)

* Fix building doc

* Fix flake8

* Support torch 1.10.x (#914)

* Support torch 1.10.x

* Fix installing PyTorch.

* Update INSTALL.rst (#915)

* Update INSTALL.rst

Setting a few additional env variables to enable compilation from source *with CUDA GPU computation support enabled*

* Fix torch/cuda/python versions in the doc. (#918)

* Fix torch/cuda/python versions in the doc.

* Minor fixes.

* Fix building for CUDA 11.6 (#917)

* Fix building for CUDA 11.6

* Minor fixes.

* Implement Unstack (#920)

* Implement unstack

* Remove code does not relate to this PR

* Remove for loop on output dim; add Unstack ragged

* Add more docs

* Fix comments

* Fix docs & unit tests

* SubsetRagged & PruneRagged (#919)

* Extend interface of SubsampleRagged.

* Add interface for pruning ragged tensor.

* Draft of new RNN-T decoding method

* Implements SubsampleRaggedShape

* Implements PruneRagged

* Rename subsample-> subset

* Minor fixes

* Fix comments

Co-authored-by: Daniel Povey <dpovey@gmail.com>

* Add Hash64 (#895)

* Add hash64

* Fix tests

* Resize hash64

* Fix comments

* fix typo

* Modified rnnt (#902)

* Add modified mutual_information_recursion

* Add modified rnnt loss

* Using more efficient way to fix boundaries

* Fix modified pruned rnnt loss

* Fix the s_begin constrains of pruned loss for modified version transducer

* Fix Stack (#925)

* return the correct layer

* unskip the test

* Fix 'TypeError' of rnnt_loss_pruned function. (#924)

* Fix 'TypeError' of rnnt_loss_simple function.

Fix 'TypeError' exception when calling rnnt_loss_simple(..., return_grad=False)  at validation steps.

* Fix 'MutualInformationRecursionFunction.forward()' return type check error for pytorch < 1.10.x

* Modify return type.

* Add documents about class MutualInformationRecursionFunction.

* Formated code style.

* Fix rnnt_loss_smoothed return type.

Co-authored-by: gzchenduisheng <gzchenduisheng@corp.netease.com>

* Support torch 1.11.0 and CUDA 11.5 (#931)

* Support torch 1.11.0 and CUDA 11.5

* Implement Rnnt decoding (#926)

* first working draft of rnnt decoding

* FormatOutput works...

* Different num frames for FormatOutput works

* Update docs

* Fix comments, break advance into several stages, add more docs

* Add python wrapper

* Add more docs

* Minor fixes

* Fix comments

* fix building docs (#933)

* Release v1.14

* Remove unused DiscountedCumSum. (#936)

* Fix compiler warnings. (#937)

* Fix compiler warnings.

* Minor fixes for RNN-T decoding. (#938)

* Minor fixes for RNN-T decoding.

* Removes arcs with label 0 from the TrivialGraph. (#939)

* Implement linear_fsa_with_self_loops. (#940)

* Implement linear_fsa_with_self_loops.

* Fix the pruning with max-states (#941)

Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>
Co-authored-by: Piotr Żelasko <petezor@gmail.com>
Co-authored-by: Jan "yenda" Trmal <jtrmal@gmail.com>
Co-authored-by: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com>
Co-authored-by: Ludwig Kürzinger <lumaku@users.noreply.github.com>
Co-authored-by: Daniel Povey <dpovey@gmail.com>
Co-authored-by: drawfish <duisheng.chen@gmail.com>
Co-authored-by: gzchenduisheng <gzchenduisheng@corp.netease.com>
Co-authored-by: alexei-v-ivanov <alexei_v_ivanov@ieee.org>
Co-authored-by: Wang, Guanbo <wgb14@outlook.com>
  • Loading branch information
11 people committed Mar 31, 2022
1 parent 0c8b470 commit 1d30b9a
Show file tree
Hide file tree
Showing 56 changed files with 4,240 additions and 1,208 deletions.
1 change: 1 addition & 0 deletions .flake8
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ per-file-ignores =
# line break before operator W503
k2/python/k2/rnnt_loss.py: E501, W503
k2/python/tests/rnnt_loss_test.py: W503
k2/python/tests/rnnt_decode_test.py: W503
exclude =
.git,
setup.py,
Expand Down
33 changes: 30 additions & 3 deletions .github/workflows/build-cpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -37,10 +37,35 @@ jobs:
fail-fast: false
matrix:
os: [ubuntu-18.04, macos-10.15]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
# Python 3.9 is for PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10.x
python-version: [3.6, 3.7, 3.8, 3.9]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
# Python 3.9 is for PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10.x, 1.11.x
# Python 3.10 is for PyTorch 1.11.x
python-version: ["3.6", "3.7", "3.8", "3.9", "3.10"]
exclude:
- python-version: "3.10" # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: "3.10"
torch: "1.5.1"
- python-version: "3.10"
torch: "1.6.0"
- python-version: "3.10"
torch: "1.7.0"
- python-version: "3.10"
torch: "1.7.1"
- python-version: "3.10"
torch: "1.8.0"
- python-version: "3.10"
torch: "1.8.1"
- python-version: "3.10"
torch: "1.9.0"
- python-version: "3.10"
torch: "1.9.1"
- python-version: "3.10"
torch: "1.10.0"
- python-version: "3.10"
torch: "1.10.1"
- python-version: "3.10"
torch: "1.10.2"
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
- python-version: 3.9
Expand All @@ -49,6 +74,8 @@ jobs:
torch: "1.6.0"
- python-version: 3.9
torch: "1.7.0"
- python-version: 3.6 # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
78 changes: 70 additions & 8 deletions .github/workflows/build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,21 +40,51 @@ jobs:
# from https://download.pytorch.org/whl/torch_stable.html
# Note: There are no torch versions for CUDA 11.2
#
# 1.11.x supports: cuda10.2 (default), 11.3, 11.5
# 1.10.x supports: cuda10.2 (default), 11.1, 11.3
# 1.9.x supports: cuda10.2 (default), 11.1
# PyTorch 1.8.x supports: cuda 10.1, 10.2 (default), 11.1
# PyTorch 1.7.x supports: cuda 10.1, 10.2 (default), 11.0
# PyTorch 1.6.0 supports: cuda 10.1, 10.2 (default)
# PyTorch 1.5.x supports: cuda 10.1, 10.2 (default)
# Other PyTorch versions are not tested
# CUDA 11.3 is for torch 1.10
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3"]
# CUDA 10.1 is for 1.5.x, 1.6.0, 1.7.x, 1.8.x
# CUDA 11.1 is for torch 1.8.x, 1.9.x, 1.10.x
# CUDA 11.3 is for torch 1.10, 1.11.x
# CUDA 11.5 is for torch 1.11.x
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3", "11.5"]
gcc: ["7"]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
#
# Python 3.9 is for PyTorch 1.7.1, 1.8.0, 1.8.1, 1.9.x, 1.10.x
python-version: [3.6, 3.7, 3.8, 3.9]
# torch 1.11.x does not support Python 3.6
# From torch 1.11.x, it supports Python 3.10
# Python 3.9 is for PyTorch 1.7.1, 1.8.0, 1.8.1, 1.9.x, 1.10.x, 11.x
python-version: ["3.6", "3.7", "3.8", "3.9", "3.10"]
exclude:
- cuda: "11.5" # exclude 11.5 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- cuda: "11.5"
torch: "1.5.1"
- cuda: "11.5"
torch: "1.6.0"
- cuda: "11.5"
torch: "1.7.0"
- cuda: "11.5"
torch: "1.7.1"
- cuda: "11.5"
torch: "1.8.0"
- cuda: "11.5"
torch: "1.8.1"
- cuda: "11.5"
torch: "1.9.0"
- cuda: "11.5"
torch: "1.9.1"
- cuda: "11.5"
torch: "1.10.0"
- cuda: "11.5"
torch: "1.10.1"
- cuda: "11.5"
torch: "1.10.2"
- cuda: "11.3" # exclude 11.3 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1]
torch: "1.5.0"
- cuda: "11.3"
Expand All @@ -73,7 +103,7 @@ jobs:
torch: "1.9.0"
- cuda: "11.3"
torch: "1.9.1"
- cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
- cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0]
torch: "1.5.0"
- cuda: "11.0"
torch: "1.5.1"
Expand All @@ -93,7 +123,9 @@ jobs:
torch: "1.10.1"
- cuda: "11.0"
torch: "1.10.2"
- cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1]
- cuda: "11.0"
torch: "1.11.0"
- cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.11.0]
torch: "1.5.0"
- cuda: "11.1"
torch: "1.5.1"
Expand All @@ -103,7 +135,9 @@ jobs:
torch: "1.7.0"
- cuda: "11.1"
torch: "1.7.1"
- cuda: "10.1" # exclude CUDA 10.1 for [1.9.0, 1.9.1, 1.10.0, 10.1, 10.2]
- cuda: "11.1"
torch: "1.11.0"
- cuda: "10.1" # exclude CUDA 10.1 for [1.9.0, 1.9.1, 1.10.0, 10.1, 10.2, 1.11.0]
torch: "1.9.0"
- cuda: "10.1"
torch: "1.9.1"
Expand All @@ -113,6 +147,8 @@ jobs:
torch: "1.10.1"
- cuda: "10.1"
torch: "1.10.2"
- cuda: "10.1"
torch: "1.11.0"
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
- python-version: 3.9
Expand All @@ -121,6 +157,32 @@ jobs:
torch: "1.6.0"
- python-version: 3.9
torch: "1.7.0"
- python-version: "3.10" # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: "3.10"
torch: "1.5.1"
- python-version: "3.10"
torch: "1.6.0"
- python-version: "3.10"
torch: "1.7.0"
- python-version: "3.10"
torch: "1.7.1"
- python-version: "3.10"
torch: "1.8.0"
- python-version: "3.10"
torch: "1.8.1"
- python-version: "3.10"
torch: "1.9.0"
- python-version: "3.10"
torch: "1.9.1"
- python-version: "3.10"
torch: "1.10.0"
- python-version: "3.10"
torch: "1.10.1"
- python-version: "3.10"
torch: "1.10.2"
- python-version: "3.6" # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
99 changes: 88 additions & 11 deletions .github/workflows/build_conda.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,10 +33,12 @@ jobs:
fail-fast: false
matrix:
os: [ubuntu-18.04]
python-version: [3.6, 3.7, 3.8, 3.9]
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3"]
python-version: ["3.6", "3.7", "3.8", "3.9", "3.10"]
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3", "11.5"]
# from https://download.pytorch.org/whl/torch_stable.html
# Note: There are no torch versions for CUDA 11.2
#
# 1.11.x supports: cuda10.2 (default), 11.3, 11.5
# PyTorch 1.10.x supports: 10.2 (default), 11.1, 11.3
# PyTorch 1.9.x supports: 10.2 (default), 11.1
# PyTorch 1.8.1 supports: cuda 10.1, 10.2 (default), 11.1
Expand All @@ -45,7 +47,8 @@ jobs:
# PyTorch 1.6.0 supports: cuda 10.1, 10.2 (default), 9.2 (not included in this setup)
# PyTorch 1.5.x supports: cuda 10.1, 10.2 (default), 9.2 (not included in this setup)
#
# PyTorch 1.7.1, 1.8.x, 1.9.x, and 1.10 support 3.6, 3.7, 3.8, 3.9
# PyTorch 1.11.x supports Python 3.10
# PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10.x, and 1.11.x support 3.6, 3.7, 3.8, 3.9
# PyTorch 1.7.0, 1.6.0, and 1.5.x support 3.6, 3.7, 3.8
#
# Other PyTorch versions are not tested
Expand All @@ -56,9 +59,51 @@ jobs:
# https://github.com/csukuangfj/k2/runs/2533830771?check_suite_focus=true
# and
# https://github.com/NVIDIA/apex/issues/805
torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
exclude:
# - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
- cuda: "11.5" # exclude cuda 11.5 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- cuda: "11.5"
torch: "1.5.1"
- cuda: "11.5"
torch: "1.6.0"
- cuda: "11.5"
torch: "1.7.0"
- cuda: "11.5"
torch: "1.7.1"
- cuda: "11.5"
torch: "1.8.0"
- cuda: "11.5"
torch: "1.8.1"
- cuda: "11.5"
torch: "1.9.0"
- cuda: "11.5"
torch: "1.9.1"
- cuda: "11.5"
torch: "1.10.0"
- cuda: "11.5"
torch: "1.10.1"
- cuda: "11.5"
torch: "1.10.2"
- cuda: "11.3" # exclude cuda 11.3 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1]
torch: "1.5.0"
- cuda: "11.3"
torch: "1.5.1"
- cuda: "11.3"
torch: "1.6.0"
- cuda: "11.3"
torch: "1.7.0"
- cuda: "11.3"
torch: "1.7.1"
- cuda: "11.3"
torch: "1.8.0"
- cuda: "11.3"
torch: "1.8.1"
- cuda: "11.3"
torch: "1.9.0"
- cuda: "11.3"
torch: "1.9.1"
# - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0]
# torch: "1.5.0"
# - cuda: "11.0"
# torch: "1.5.1"
Expand All @@ -78,7 +123,9 @@ jobs:
torch: "1.10.1"
- cuda: "11.0"
torch: "1.10.2"
# - cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1]
- cuda: "11.0"
torch: "1.11.0"
# - cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.11.0]
# torch: "1.5.0"
# - cuda: "11.1"
# torch: "1.5.1"
Expand All @@ -88,7 +135,9 @@ jobs:
torch: "1.7.0"
- cuda: "11.1"
torch: "1.7.1"
- cuda: "10.1" # exclude 10.1 for [1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
- cuda: "11.1"
torch: "1.11.0"
- cuda: "10.1" # exclude 10.1 for [1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0]
torch: "1.9.0"
- cuda: "10.1"
torch: "1.9.1"
Expand All @@ -98,14 +147,42 @@ jobs:
torch: "1.10.1"
- cuda: "10.1"
torch: "1.10.2"
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
- cuda: "10.1"
torch: "1.11.0"
- python-version: "3.9" # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
- python-version: "3.9"
torch: "1.5.1"
- python-version: "3.9"
torch: "1.6.0"
- python-version: "3.9"
torch: "1.7.0"
- python-version: "3.10" # exclude Python 3.10 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: 3.9
- python-version: "3.10"
torch: "1.5.1"
- python-version: 3.9
- python-version: "3.10"
torch: "1.6.0"
- python-version: 3.9
- python-version: "3.10"
torch: "1.7.0"
- python-version: "3.10"
torch: "1.7.1"
- python-version: "3.10"
torch: "1.8.0"
- python-version: "3.10"
torch: "1.8.1"
- python-version: "3.10"
torch: "1.9.0"
- python-version: "3.10"
torch: "1.9.1"
- python-version: "3.10"
torch: "1.10.0"
- python-version: "3.10"
torch: "1.10.1"
- python-version: "3.10"
torch: "1.10.2"
- python-version: "3.6" # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
39 changes: 33 additions & 6 deletions .github/workflows/build_conda_cpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -43,24 +43,51 @@ jobs:
fail-fast: false
matrix:
os: [ubuntu-18.04, macos-10.15]
python-version: [3.6, 3.7, 3.8, 3.9]
python-version: ["3.6", "3.7", "3.8", "3.9", "3.10"]
# from https://download.pytorch.org/whl/torch_stable.html
#
# PyTorch 1.11.x supports 3.7, 3.8, 3.9, 3.10
# PyTorch 1.10, 1.9.x, 1.8.x, and 1.7.1 support 3.6, 3.7, 3.8, 3.9
# PyTorch 1.7.0, 1.6.0, and 1.5.x support 3.6, 3.7, 3.8
#
# Other PyTorch versions are not tested
#
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
exclude:
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
- python-version: "3.9" # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
- python-version: 3.9
- python-version: "3.9"
torch: "1.5.1"
- python-version: 3.9
- python-version: "3.9"
torch: "1.6.0"
- python-version: 3.9
- python-version: "3.9"
torch: "1.7.0"
- python-version: "3.10" # exclude Python 3.10 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: "3.10"
torch: "1.5.1"
- python-version: "3.10"
torch: "1.6.0"
- python-version: "3.10"
torch: "1.7.0"
- python-version: "3.10"
torch: "1.7.1"
- python-version: "3.10"
torch: "1.8.0"
- python-version: "3.10"
torch: "1.8.1"
- python-version: "3.10"
torch: "1.9.0"
- python-version: "3.10"
torch: "1.9.1"
- python-version: "3.10"
torch: "1.10.0"
- python-version: "3.10"
torch: "1.10.1"
- python-version: "3.10"
torch: "1.10.2"
- python-version: "3.6" # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
Loading

0 comments on commit 1d30b9a

Please sign in to comment.