Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Npu dev #2385

Merged
merged 45 commits into from
Nov 3, 2022
Merged

Npu dev #2385

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
45 commits
Select commit Hold shift + click to select a range
83bedf5
[Feature] Support RoipointPool3d with cambricon MLU backend (#2247)
ZShaopeng Sep 8, 2022
3a311e8
[Docs] Add zhihu articles (#2157)
zhouzaida Sep 11, 2022
2fb2b91
Add type hints for mmcv/image (#2089)
triple-Mu Sep 12, 2022
2046a39
[Fix] Use ROCm backened within the PyTorch framework (#1918)
zstreet87 Sep 13, 2022
55c51e1
[Feature] Register a new activatation layer SiLU to ACTIVATION_LAYERS…
okotaku Sep 13, 2022
25f533b
Fix spconv clang-format (#2260)
Sep 14, 2022
3b9b905
[Enhancement] Add support of TorchVision's Model Registration API on …
datumbox Sep 14, 2022
629ef89
Pick commits from 2.x branch (#2284)
zhouzaida Sep 25, 2022
847612d
Bump version to v1.6.2 (#2285)
zhouzaida Sep 26, 2022
afff388
[Enhancement] Support SiLU with torch < 1.7.0 (#2278)
okotaku Sep 29, 2022
9250417
Update onnx registry (#2294)
Sep 29, 2022
a5db5f6
[Feature] Support training on NPU device (#2262)
wangjiangben-hw Sep 30, 2022
b6a7fd9
Upgrade pre commit hooks (#2321)
zhouzaida Oct 8, 2022
8e844d1
Fix warning of CUDA ops (#2324)
tpoisonooo Oct 10, 2022
b1711db
[Refactor] Adapt mlu code to cntoolkit3.0.1 (#2248)
defei-coder Oct 10, 2022
c001e2f
[Refactor] Refactor the nms3d op to support MLU (#2296)
Oct 11, 2022
dd5b415
[Feature] Support PrRoI op for Parrots (#2280)
CokeDong Oct 11, 2022
e417035
[Enhancement] Add ability to pass logger instance to frameworks (#2317)
shcheklein Oct 12, 2022
bd1da5a
[Docs] Refine README and installation documentations (#1977)
zhouzaida Oct 17, 2022
be684ee
[Feature] Add getJobLimitCapability interface and use it in nms (#2337)
defei-coder Oct 18, 2022
04ec054
[Fix] Fix the bug of uninitialized prarmeter (#2345)
defei-coder Oct 20, 2022
6cf9ac0
[Fix] Fix three nn op can not accept half tensor (#2348)
Oct 24, 2022
6d9e4a9
Build macOS wheel in Github Action (#2311)
zhouzaida Oct 24, 2022
7fd7058
[Feature] Support Iou3d with cambricon MLU backend (#2339)
cathyzhang222 Oct 24, 2022
ff18904
[Fix] Fix Correlation op (#2274)
Oct 24, 2022
9709ff3
[Enhancement] Add a new argument define_metric in wandb hook (#2237)
okotaku Oct 24, 2022
3c35bab
[Fix] Fix the potential NaN bug in calc_square_dist() (#2356)
ZCMax Oct 26, 2022
1eef4f5
[Docs] Update README and fix typo in docs (#2358)
zhouzaida Oct 26, 2022
f48975b
[Fix] Fix MPS ops for torch1.12+ (#2302)
grimoire Sep 29, 2022
a4f304a
[Feature] Add box_iou_quadri & nms_quadri (#2277)
zytx121 Oct 13, 2022
93fe482
[Features] Add stack ball query and stack group points ops (#2292)
VVsssssk Oct 27, 2022
b513419
Build torch1.13 pre-built package for macOS in CI (#2366)
zhouzaida Oct 31, 2022
14e8e49
[Fix] Fix set_device bug when using multi-machine multi-device (#2370)
wangjiangben-hw Oct 31, 2022
e2c9fc4
[Docs] Add steps to install mmcv-full on NPU machine (#2363)
momo609 Oct 31, 2022
ff8e0c8
[Docs] Add maintenance plan and print deprecated message (#2357)
zhouzaida Nov 1, 2022
270c293
Bump version to v1.7.0 (#2362)
zhouzaida Nov 1, 2022
b622fb2
[Feature] Support RoIAwarePool3d with cambricon MLU backend (#2359)
ZhangLearning Nov 3, 2022
ff3ffbb
init npu
wangjiangben-hw Sep 14, 2022
57e4171
add npu extension and focal loss adapter
ckirchhoff2021 Sep 19, 2022
65b5fe1
clean code
ckirchhoff2021 Sep 20, 2022
f38b1b2
clean code
ckirchhoff2021 Sep 20, 2022
f5c156f
clean code
ckirchhoff2021 Sep 20, 2022
c5ffc60
add modulatedDeformConv npu adapter
ckirchhoff2021 Nov 3, 2022
bd3ec11
merge master branch 20221103
ckirchhoff2021 Nov 3, 2022
74d3181
remove conflicts
ckirchhoff2021 Nov 3, 2022
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 2 additions & 6 deletions .github/workflows/build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -157,10 +157,8 @@ jobs:
strategy:
matrix:
python-version: [3.7]
torch: [1.3.1, 1.5.1+cu101, 1.6.0+cu101, 1.7.0+cu101, 1.8.0+cu101]
torch: [1.5.1+cu101, 1.6.0+cu101, 1.7.0+cu101, 1.8.0+cu101]
include:
- torch: 1.3.1
torchvision: 0.4.2
- torch: 1.5.1+cu101
torchvision: 0.6.1+cu101
- torch: 1.6.0+cu101
Expand Down Expand Up @@ -362,10 +360,8 @@ jobs:
runs-on: macos-latest
strategy:
matrix:
torch: [1.3.1, 1.5.1, 1.6.0, 1.7.0, 1.8.0, 1.9.0]
torch: [1.5.1, 1.6.0, 1.7.0, 1.8.0, 1.9.0]
include:
- torch: 1.3.1
torchvision: 0.4.2
- torch: 1.5.1
torchvision: 0.6.1
- torch: 1.6.0
Expand Down
81 changes: 81 additions & 0 deletions .github/workflows/build_macos_wheel.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
name: build macos wheel

on: push

concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true

env:
MMCV_WITH_OPS: 1

jobs:
build_macos10_wheel:
runs-on: macos-10.15
if: contains(github.event.head_commit.message, 'Bump version to')
strategy:
matrix:
torch: [1.5.1, 1.6.0, 1.7.0, 1.8.0, 1.9.0, 1.10.0, 1.11.0, 1.12.0, 1.13.0]
python-version: [3.6, 3.7, 3.8, 3.9, '3.10']
include:
- torch: 1.5.1
torchvision: 0.6.1
- torch: 1.6.0
torchvision: 0.7.0
- torch: 1.7.0
torchvision: 0.8.1
- torch: 1.8.0
torchvision: 0.9.0
- torch: 1.9.0
torchvision: 0.10.0
- torch: 1.10.0
torchvision: 0.11.0
- torch: 1.11.0
torchvision: 0.12.0
- torch: 1.12.0
torchvision: 0.13.0
- torch: 1.13.0
torchvision: 0.14.0
exclude:
- torch: 1.5.1
python-version: 3.9
- torch: 1.5.1
python-version: '3.10'
- torch: 1.6.0
python-version: 3.9
- torch: 1.6.0
python-version: '3.10'
- torch: 1.7.0
python-version: 3.9
- torch: 1.7.0
python-version: '3.10'
- torch: 1.8.0
python-version: '3.10'
- torch: 1.9.0
python-version: '3.10'
- torch: 1.10.0
python-version: '3.10'
- torch: 1.11.0
python-version: 3.6
- torch: 1.12.0
python-version: 3.6
- torch: 1.13.0
python-version: 3.6
steps:
- uses: actions/checkout@v2
- name: Set up Python
uses: actions/setup-python@v2
with:
python-version: ${{ matrix.python-version }}
- name: Install psutil
run: pip install psutil
- name: Install PyTorch
run: pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} --no-cache-dir
- name: Build and install
run: |
pip install wheel
python setup.py bdist_wheel
- uses: actions/upload-artifact@v3
with:
name: ${{matrix.torch}}
path: dist/
10 changes: 5 additions & 5 deletions .pre-commit-config-zh-cn.yaml
Original file line number Diff line number Diff line change
@@ -1,19 +1,19 @@
exclude: ^tests/data/
repos:
- repo: https://gitee.com/openmmlab/mirrors-flake8
rev: 3.8.3
rev: 5.0.4
hooks:
- id: flake8
- repo: https://gitee.com/openmmlab/mirrors-isort
rev: 5.10.1
hooks:
- id: isort
- repo: https://gitee.com/openmmlab/mirrors-yapf
rev: v0.30.0
rev: v0.32.0
hooks:
- id: yapf
- repo: https://gitee.com/openmmlab/mirrors-pre-commit-hooks
rev: v3.1.0
rev: v4.3.0
hooks:
- id: trailing-whitespace
- id: check-yaml
Expand All @@ -26,7 +26,7 @@ repos:
- id: mixed-line-ending
args: ["--fix=lf"]
- repo: https://gitee.com/openmmlab/mirrors-codespell
rev: v2.1.0
rev: v2.2.1
hooks:
- id: codespell
- repo: https://gitee.com/openmmlab/mirrors-mdformat
Expand All @@ -44,7 +44,7 @@ repos:
- id: docformatter
args: ["--in-place", "--wrap-descriptions", "79"]
- repo: https://github.com/asottile/pyupgrade
rev: v2.32.1
rev: v3.0.0
hooks:
- id: pyupgrade
args: ["--py36-plus"]
Expand Down
11 changes: 5 additions & 6 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
@@ -1,19 +1,19 @@
exclude: ^tests/data/
repos:
- repo: https://github.com/PyCQA/flake8
rev: 3.8.3
rev: 5.0.4
hooks:
- id: flake8
- repo: https://github.com/PyCQA/isort
rev: 5.10.1
hooks:
- id: isort
- repo: https://github.com/pre-commit/mirrors-yapf
rev: v0.30.0
rev: v0.32.0
hooks:
- id: yapf
- repo: https://github.com/pre-commit/pre-commit-hooks
rev: v3.1.0
rev: v4.3.0
hooks:
- id: trailing-whitespace
- id: check-yaml
Expand All @@ -26,10 +26,9 @@ repos:
- id: mixed-line-ending
args: ["--fix=lf"]
- repo: https://github.com/codespell-project/codespell
rev: v2.1.0
rev: v2.2.1
hooks:
- id: codespell
exclude: ^README.md
- repo: https://github.com/executablebooks/mdformat
rev: 0.7.9
hooks:
Expand All @@ -45,7 +44,7 @@ repos:
- id: docformatter
args: ["--in-place", "--wrap-descriptions", "79"]
- repo: https://github.com/asottile/pyupgrade
rev: v2.32.1
rev: v3.0.0
hooks:
- id: pyupgrade
args: ["--py36-plus"]
Expand Down
Loading