mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
fix build ci not triggerred (#578)
* fix ci * ignore scripts * remove pycuda from requirements * fix trt apis
This commit is contained in:
parent
c99379cc76
commit
b29a8a020a
7
.github/workflows/build.yml
vendored
7
.github/workflows/build.yml
vendored
@ -3,12 +3,16 @@ name: build
|
|||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
paths-ignore:
|
paths-ignore:
|
||||||
|
- ".github/scripts/**"
|
||||||
- "demo/**"
|
- "demo/**"
|
||||||
|
- "docker/**"
|
||||||
- "tools/**"
|
- "tools/**"
|
||||||
|
|
||||||
pull_request:
|
pull_request:
|
||||||
paths-ignore:
|
paths-ignore:
|
||||||
|
- ".github/scripts/**"
|
||||||
- "demo/**"
|
- "demo/**"
|
||||||
|
- "docker/**"
|
||||||
- "tools/**"
|
- "tools/**"
|
||||||
- "docs/**"
|
- "docs/**"
|
||||||
|
|
||||||
@ -89,6 +93,7 @@ jobs:
|
|||||||
python -V
|
python -V
|
||||||
python -m pip install mmcv-full==${{matrix.mmcv}} -f https://download.openmmlab.com/mmcv/dist/cu102/${{matrix.torch_version}}/index.html
|
python -m pip install mmcv-full==${{matrix.mmcv}} -f https://download.openmmlab.com/mmcv/dist/cu102/${{matrix.torch_version}}/index.html
|
||||||
python -m pip install -r requirements.txt
|
python -m pip install -r requirements.txt
|
||||||
|
pip install -U pycuda
|
||||||
python -m pip install -U numpy
|
python -m pip install -U numpy
|
||||||
- name: Build and install
|
- name: Build and install
|
||||||
run: |
|
run: |
|
||||||
@ -134,6 +139,7 @@ jobs:
|
|||||||
python -V
|
python -V
|
||||||
python -m pip install mmcv-full==${{matrix.mmcv}} -f https://download.openmmlab.com/mmcv/dist/cu111/${{matrix.torch_version}}/index.html
|
python -m pip install mmcv-full==${{matrix.mmcv}} -f https://download.openmmlab.com/mmcv/dist/cu111/${{matrix.torch_version}}/index.html
|
||||||
python -m pip install -r requirements.txt
|
python -m pip install -r requirements.txt
|
||||||
|
pip install -U pycuda
|
||||||
python -m pip install -U numpy
|
python -m pip install -U numpy
|
||||||
- name: Build and install
|
- name: Build and install
|
||||||
run: |
|
run: |
|
||||||
@ -152,4 +158,3 @@ jobs:
|
|||||||
env_vars: OS,PYTHON
|
env_vars: OS,PYTHON
|
||||||
name: codecov-umbrella
|
name: codecov-umbrella
|
||||||
fail_ci_if_error: false
|
fail_ci_if_error: false
|
||||||
gcov_ignore : [".github/scripts/*"]
|
|
||||||
|
@ -1,16 +1,14 @@
|
|||||||
# Copyright (c) OpenMMLab. All rights reserved.
|
# Copyright (c) OpenMMLab. All rights reserved.
|
||||||
from mmdeploy.backend.tensorrt import from_onnx as _from_onnx
|
from mmdeploy.backend.tensorrt import is_available, is_custom_ops_available
|
||||||
from mmdeploy.backend.tensorrt import (is_available, is_custom_ops_available,
|
|
||||||
load, save)
|
|
||||||
from ..core import PIPELINE_MANAGER
|
from ..core import PIPELINE_MANAGER
|
||||||
|
|
||||||
from_onnx = PIPELINE_MANAGER.register_pipeline()(_from_onnx)
|
__all__ = ['is_available', 'is_custom_ops_available']
|
||||||
|
|
||||||
__all__ = [
|
|
||||||
'is_available', 'is_custom_ops_available', 'from_onnx', 'save', 'load'
|
|
||||||
]
|
|
||||||
|
|
||||||
if is_available():
|
if is_available():
|
||||||
|
from mmdeploy.backend.tensorrt import from_onnx as _from_onnx
|
||||||
|
from mmdeploy.backend.tensorrt import load, save
|
||||||
|
from_onnx = PIPELINE_MANAGER.register_pipeline()(_from_onnx)
|
||||||
|
__all__ += ['from_onnx', 'save', 'load']
|
||||||
try:
|
try:
|
||||||
from mmdeploy.backend.tensorrt.onnx2tensorrt import \
|
from mmdeploy.backend.tensorrt.onnx2tensorrt import \
|
||||||
onnx2tensorrt as _onnx2tensorrt
|
onnx2tensorrt as _onnx2tensorrt
|
||||||
|
@ -7,4 +7,3 @@ mmrazor>=0.3.0
|
|||||||
mmsegmentation
|
mmsegmentation
|
||||||
onnxruntime>=1.8.0
|
onnxruntime>=1.8.0
|
||||||
openvino-dev
|
openvino-dev
|
||||||
pycuda
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user