Overview
Our example tests and documentation tests use torch 1.12.0, see
|
image: hpcaitech/pytorch-cuda:1.12.0-11.3.0 |
.
However, out unit tests use torch 1.11.0, see
|
image: hpcaitech/pytorch-cuda:1.11.0-11.3.0 |
.
This inconsistency may lead to compatibility issue.
Goal
- Update torch verion of CI to 1.12.0
- Fix failed tests.
Overview
Our example tests and documentation tests use torch 1.12.0, see
ColossalAI/.github/workflows/example_check_on_pr.yml
Line 75 in f7361ee
However, out unit tests use torch 1.11.0, see
ColossalAI/.github/workflows/build_on_pr.yml
Line 71 in f7361ee
This inconsistency may lead to compatibility issue.
Goal