fix build ci not triggerred (#578)

* fix ci

* ignore scripts

* remove pycuda from requirements

* fix trt apis
This commit is contained in:
RunningLeon 2022-06-09 11:19:05 +08:00 committed by GitHub
parent c99379cc76
commit b29a8a020a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 10 deletions

View File

@ -3,12 +3,16 @@ name: build
on:
push:
paths-ignore:
- ".github/scripts/**"
- "demo/**"
- "docker/**"
- "tools/**"
pull_request:
paths-ignore:
- ".github/scripts/**"
- "demo/**"
- "docker/**"
- "tools/**"
- "docs/**"
@ -89,6 +93,7 @@ jobs:
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 -r requirements.txt
pip install -U pycuda
python -m pip install -U numpy
- name: Build and install
run: |
@ -134,6 +139,7 @@ jobs:
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 -r requirements.txt
pip install -U pycuda
python -m pip install -U numpy
- name: Build and install
run: |
@ -152,4 +158,3 @@ jobs:
env_vars: OS,PYTHON
name: codecov-umbrella
fail_ci_if_error: false
gcov_ignore : [".github/scripts/*"]

View File

@ -1,16 +1,14 @@
# 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,
load, save)
from mmdeploy.backend.tensorrt import is_available, is_custom_ops_available
from ..core import PIPELINE_MANAGER
from_onnx = PIPELINE_MANAGER.register_pipeline()(_from_onnx)
__all__ = [
'is_available', 'is_custom_ops_available', 'from_onnx', 'save', 'load'
]
__all__ = ['is_available', 'is_custom_ops_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:
from mmdeploy.backend.tensorrt.onnx2tensorrt import \
onnx2tensorrt as _onnx2tensorrt

View File

@ -7,4 +7,3 @@ mmrazor>=0.3.0
mmsegmentation
onnxruntime>=1.8.0
openvino-dev
pycuda