Try again

Signed-off-by: lizz <lizz@sensetime.com>
pull/43/head
lizz 2021-04-10 17:10:49 +08:00
parent 2341bc5528
commit ee1dbf1037
2 changed files with 1 additions and 5 deletions

View File

@ -135,10 +135,7 @@ jobs:
run: |
rm -rf .eggs
python setup.py check -m -s
echo $CUDA_HOME
ls /usr/local/
python -c 'import torch;print(torch.cuda.is_available())'
TORCH_CUDA_ARCH_LIST=7.0 pip install .
TORCH_CUDA_ARCH_LIST=7.0 pip install -e .
- name: Run unittests and generate coverage report
run: |
coverage run --branch --source mmocr -m pytest tests/

View File

@ -102,7 +102,6 @@ def parse_requirements(fname='requirements.txt', with_version=True):
def get_rroi_align_extensions():
print('XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX')
extensions_dir = 'mmocr/models/utils/ops/rroi_align/csrc/csc'
main_file = glob.glob(os.path.join(extensions_dir, '*.cpp'))
source_cpu = glob.glob(os.path.join(extensions_dir, 'cpu', '*.cpp'))