1. 05 Sep, 2023 4 commits
  2. 13 Jun, 2023 1 commit
  3. 19 May, 2023 1 commit
  4. 19 Apr, 2023 1 commit
    • bdf's avatar
      Pick MLU modifications from master (1.x) to main (2.x) (#2704) · 733e6ff8
      bdf authored
      
      
      * [Feature] Support Voxelization with cambricon MLU device (#2500)
      
      * [Feature] Support hard_voxelize with cambricon MLU backend
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Feature](bangc-ops): add voxelization op
      
      * [Enhance] Optimize the performace of ms_deform_attn for MLU device (#2510)
      
      * ms_opt
      
      * ms_opt
      
      * ms_opt
      
      * ms_opt
      
      * ms_opt
      
      * [Feature] ms_deform_attn performance optimization
      
      * [Feature] ms_deform_attn performance optimization
      
      * [Feature] ms_deform_attn performance optimization
      
      * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520)
      
      * [Feature] Support ball_query with cambricon MLU backend and mlu-ops library.
      
      * [Fix] update operator data layout setting.
      
      * [Fix] add cxx compile option to avoid symbol conflict.
      
      * [Fix] fix lint errors.
      
      * [Fix] update ops.md with info of ball_query support by MLU backend.
      
      * [Feature] Fix typo.
      
      * [Fix] Remove print.
      
      * [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env.
      
      * [Fix] update MMCV_MLU_OPS_PATH check logic.
      
      * [Fix] update error info when failed to download mlu-ops.
      
      * [Fix] check mlu-ops version matching info in mmcv.
      
      * [Fix] revise wrong filename.
      
      * [Fix] remove f.close and re.
      
      * [Docs] Steps to compile mmcv-full on MLU machine (#2571)
      
      * [Docs] Steps to compile mmcv-full on MLU machine
      
      * [Docs] Adjust paragraph order
      
      * Update docs/zh_cn/get_started/build.md
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * Update docs/zh_cn/get_started/build.md
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * Update docs/en/get_started/build.md
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * Update docs/en/get_started/build.md
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * [Docs] Modify the format
      
      ---------
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * [Fix] Fix tensor descriptor setting in MLU ball_query. (#2579)
      
      * [Feature] Add MLU support for Sparse Convolution op (#2589)
      
      * [Feature] Add sparse convolution MLU API
      
      * [Feature] update cpp code style
      
      * end-of-file
      
      * delete libext.a
      
      * code style
      
      * update ops.md
      
      ---------
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      
      * [Enhancement] Replace the implementation of deform_roi_pool with mlu-ops (#2598)
      
      * [Feature] Replace the implementation of deform_roi_pool with mlu-ops
      
      * [Feature] Modify code
      
      ---------
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      
      * [Enhancement] ms_deform_attn performance optimization (#2616)
      
      * ms_opt_v2
      
      * ms_opt_v2_1
      
      * optimize MultiScaleDeformableAttention ops for MLU
      
      * ms_opt_v2_1
      
      * [Feature] ms_deform_attn performance optimization V2
      
      * [Feature] ms_deform_attn performance optimization V2
      
      * [Feature] ms_deform_attn performance optimization V2
      
      * [Feature] ms_deform_attn performance optimization V2
      
      * [Feature] ms_deform_attn performance optimization V2
      
      * [Feature] ms_deform_attn performance optimization V2
      
      * [Feature] ms_deform_attn performance optimization V2
      
      ---------
      Co-authored-by: default avatardongchengwei <dongchengwei@cambricon.com>
      
      * [Feature] Support NmsRotated with cambricon MLU backend (#2643)
      
      * [Feature] Support NmsRotated with cambricon MLU backend
      
      * [Feature] remove foolproofs in nms_rotated_mlu.cpp
      
      * [Feature] fix lint in test_nms_rotated.py
      
      * [Feature] fix kMLU not found in nms_rotated.cpp
      
      * [Feature] modify mlu support in nms.py
      
      * [Feature] modify nms_rotated support in ops.md
      
      * [Feature] modify ops/nms.py
      
      * [Enhance] Add a default value for MMCV_MLU_ARGS (#2688)
      
      * add mlu_args
      
      * add mlu_args
      
      * Modify the code
      
      ---------
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      
      * [Enhance] Ignore mlu-ops files (#2691)
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      
      ---------
      Co-authored-by: default avatarZShaopeng <108382403+ZShaopeng@users.noreply.github.com>
      Co-authored-by: default avatarBinZheng <38182684+Wickyzheng@users.noreply.github.com>
      Co-authored-by: default avatarliuduanhui <103939338+DanieeelLiu@users.noreply.github.com>
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      Co-authored-by: default avatarduzekun <108381389+duzekunKTH@users.noreply.github.com>
      Co-authored-by: default avatardongchengwei <dongchengwei@cambricon.com>
      Co-authored-by: default avatarliuyuan1-v <125547457+liuyuan1-v@users.noreply.github.com>
      733e6ff8
  5. 06 Apr, 2023 1 commit
  6. 25 Mar, 2023 1 commit
  7. 13 Jan, 2023 1 commit
    • ckirchhoff's avatar
      [Feature] Pick npu ops from master to 2.x (#2501) · 34bdf448
      ckirchhoff authored
      * merge npu ops from master to 2.x
      
      * BugFix: fix merge bugs
      
      * {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition
      
      * merge conflicts in ops.md
      
      * [fix]: fix merge bug
      34bdf448
  8. 06 Jan, 2023 1 commit
  9. 22 Oct, 2022 1 commit
  10. 30 Aug, 2022 1 commit
  11. 26 Aug, 2022 1 commit
  12. 03 Aug, 2022 1 commit
  13. 18 May, 2022 1 commit
  14. 18 Apr, 2022 1 commit
  15. 16 Apr, 2022 2 commits
    • Qiza-lyhm's avatar
      [Fix] Add MLU kernels to be compiled (#1885) · 42e7e2ee
      Qiza-lyhm authored
      - Add MLU operator kernels.
      42e7e2ee
    • Jiazhen Wang's avatar
      [Feature] Add several MLU ops (#1563) · 362a90f8
      Jiazhen Wang authored
      
      
      * [Feature] Add roiaware pool3d ops from mmdet3d (#1382)
      
      * add ops (roiaware pool3d) in mmdet3d
      
      * refactor code
      
      * fix typo
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Feature] Add iou3d op from mmdet3d (#1356)
      
      * add ops (iou3d) in mmdet3d
      
      * add unit test
      
      * refactor code
      
      * refactor code
      
      * refactor code
      
      * refactor code
      
      * refactor code
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Fix] Update test data for test_iou3d (#1427)
      
      * Update test data for test_iou3d
      
      * delete blank lines
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * [Feature] Add group points ops from mmdet3d (#1415)
      
      * add op (group points) and its related ops (ball query and knn) in mmdet3d
      
      * refactor code
      
      * fix typo
      
      * refactor code
      
      * fix typo
      
      * refactor code
      
      * make input contiguous
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * add mmdet3d op (#1425)
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Feature] Loading objects from different backends and dumping objects to different backends (#1330)
      
      * [Feature] Choose storage backend by the prefix of filepath
      
      * refactor FileClient and add unittest
      
      * support loading from different backends
      
      * polish docstring
      
      * fix unittet
      
      * rename attribute str_like_obj to is_str_like_obj
      
      * add infer_client method
      
      * add check_exist method
      
      * rename var client to file_client
      
      * polish docstring
      
      * add join_paths method
      
      * remove join_paths and add _format_path
      
      * enhance unittest
      
      * refactor unittest
      
      * singleton pattern
      
      * fix test_clientio.py
      
      * deprecate CephBackend
      
      * enhance docstring
      
      * refactor unittest for petrel
      
      * refactor unittest for disk backend
      
      * update io.md
      
      * add concat_paths method
      
      * improve docstring
      
      * improve docstring
      
      * add isdir and copyfile for file backend
      
      * delete copyfile and add get_local_path
      
      * remove isdir method of petrel
      
      * fix typo
      
      * add comment and polish docstring
      
      * polish docstring
      
      * rename _path_mapping to _map_path
      
      * polish docstring and fix typo
      
      * refactor get_local_path
      
      * add list_dir_or_file for FileClient
      
      * add list_dir_or_file for PetrelBackend
      
      * fix windows ci
      
      * Add return docstring
      
      * polish docstring
      
      * fix typo
      
      * fix typo
      
      * deprecate the conversion from Path to str
      
      * add docs for loading checkpoints with FileClient
      
      * refactor map_path
      
      * add _ensure_methods to ensure methods have been implemented
      
      * fix list_dir_or_file
      
      * rename _ensure_method_implemented to has_method
      
      * Add CI for pytorch 1.10 (#1431)
      
      * [Feature] Upload checkpoints and logs to ceph (#1375)
      
      * [Feature] Choose storage backend by the prefix of filepath
      
      * refactor FileClient and add unittest
      
      * support loading from different backends
      
      * polish docstring
      
      * fix unittet
      
      * rename attribute str_like_obj to is_str_like_obj
      
      * [Docs] Upload checkpoint to petrel oss
      
      * add infer_client method
      
      * Support uploading checkpoint to petrel oss
      
      * add check_exist method
      
      * refactor CheckpointHook
      
      * support uploading logs to ceph
      
      * rename var client to file_client
      
      * polish docstring
      
      * enhance load_from_ceph
      
      * refactor load_from_ceph
      
      * refactor TextLoggerHook
      
      * change the meaning of out_dir argument
      
      * fix test_checkpoint_hook.py
      
      * add join_paths method
      
      * remove join_paths and add _format_path
      
      * enhance unittest
      
      * refactor unittest
      
      * add a unittest for EvalHook when file backend is petrel
      
      * singleton pattern
      
      * fix test_clientio.py
      
      * deprecate CephBackend
      
      * add warning in load_from_ceph
      
      * fix type of out_suffix
      
      * enhance docstring
      
      * refactor unittest for petrel
      
      * refactor unittest for disk backend
      
      * update io.md
      
      * add concat_paths method
      
      * fix CI
      
      * mock check_exist
      
      * improve docstring
      
      * improve docstring
      
      * improve docstring
      
      * improve docstring
      
      * add isdir and copyfile for file backend
      
      * delete copyfile and add get_local_path
      
      * remove isdir method of petrel
      
      * fix typo
      
      * rename check_exists to exists
      
      * refactor code and polish docstring
      
      * fix windows ci
      
      * add comment and polish docstring
      
      * polish docstring
      
      * polish docstring
      
      * rename _path_mapping to _map_path
      
      * polish docstring and fix typo
      
      * refactor get_local_path
      
      * add list_dir_or_file for FileClient
      
      * add list_dir_or_file for PetrelBackend
      
      * fix windows ci
      
      * Add return docstring
      
      * polish docstring
      
      * fix typo
      
      * fix typo
      
      * fix typo
      
      * fix error when mocking PetrelBackend
      
      * deprecate the conversion from Path to str
      
      * add docs for loading checkpoints with FileClient
      
      * rename keep_log to keep_local
      
      * refactor map_path
      
      * add _ensure_methods to ensure methods have been implemented
      
      * fix list_dir_or_file
      
      * rename _ensure_method_implemented to has_method
      
      * refactor
      
      * polish information
      
      * format information
      
      * bump version to v1.3.16 (#1430)
      
      * [Fix]: Update test data of test_tin_shift (#1426)
      
      * Update test data of test_tin_shift
      
      * Delete tmp.engine
      
      * add pytest raises asserterror test
      
      * raise valueerror, update test log
      
      * add more comment
      
      * Apply suggestions from code review
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * fix the wrong function reference bug in BaseTransformerLayer when batch_first is True (#1418)
      
      * [Docs] Add mmcv itself in the docs list (#1441)
      
      * Add mmcv itself in the docs list
      
      * modify link of docs
      
      * [Improve] improve checkpoint loading log (#1446)
      
      * [Feature] Support SigmoidFocalLoss with Cambricon MLU backend (#1346)
      
      * [Feature] Support SigmoidFocalLoss with Cambricon MLU backend
      
      * refactor MMCV_WITH_MLU macro define
      
      * refactor NFU_ALIGN_SIZE, PAD_DOWN and split_pipeline_num
      
      * delete extra fool proofing in cpp
      
      * [Feature] Support SigmoidFocalLossBackward with Cambricon MLU backend
      
      * fix macro definition in SigmoidFocalLoss
      
      * refactor mlu files into clang-format
      
      * refactor sigmoid focal loss test
      
      * refactor Sigmoid Focal Loss file structure.
      
      * fix python lint error
      
      * fix import torch_mlu error type
      
      * fix lint
      
      * refactor clang format style to google
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      
      * [Feature] Support RoiAlign With Cambricon MLU Backend (#1429)
      
      * [Feature] Support NMS with cambricon MLU backend (#1467)
      
      * [Feature] Support BBoxOverlaps with cambricon MLU backend (#1507)
      
      * [Refactor] Format C++ code
      
      * [Refactor] include common_mlu_helper in pytorch_mlu_helper and refactor build condition
      
      * [Improve] Improve the performance of roialign, nms and focalloss with MLU backend (#1572)
      
      * [Improve] Improve the performance of roialign with MLU backend
      
      * replace CHECK_MLU with CHECK_MLU_INPUT
      
      * [Improve] Improve the perf of nms and focallosssigmoid with MLU backend
      
      * [Improve] Improve the performance of roialign with MLU backend (#1741)
      
      * [Feature] Support tin_shift with cambricon MLU backend (#1696)
      
      * [Feature] Support tin_shift with cambricon MLU backend
      
      * [fix] Add the assertion of batch_size in tin_shift.py
      
      * [fix] fix the param check of tin_shift in cambricon code
      
      * [fix] Fix lint failure.
      
      * [fix] Fix source file lint failure.
      
      * Update mmcv/ops/tin_shift.py
      
      [Refactor] Modify the code in mmcv/ops/tin_shift.py.
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      Co-authored-by: default avatarbudefei <budefei@cambricom.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      
      * resolve conflicts and fix lint
      
      * fix mmcv.utils.__init__
      
      * fix mmcv.utils.__init__
      
      * Fix lints and change FLAG
      
      * fix setup and refine
      
      * remove a redundant line
      
      * remove an unnecessary 'f'
      
      * fix compilation error
      Co-authored-by: default avatardingchang <hudingchang.vendor@sensetime.com>
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      Co-authored-by: default avatarq.yao <yaoqian@sensetime.com>
      Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
      Co-authored-by: default avatarpc <luopeichao@sensetime.com>
      Co-authored-by: default avatarWenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
      Co-authored-by: default avatarq.yao <streetyao@live.com>
      Co-authored-by: default avatarTong Gao <gaotongxiao@gmail.com>
      Co-authored-by: default avatarYuxin Liu <liuyuxin@cambricon.com>
      Co-authored-by: default avatarzihanchang11 <92860914+zihanchang11@users.noreply.github.com>
      Co-authored-by: default avatarshlrao <shenglong.rao@gmail.com>
      Co-authored-by: default avatarzhouchenyang <zcy19950525@gmail.com>
      Co-authored-by: default avatarMrxiaofei <36697723+Mrxiaofei@users.noreply.github.com>
      Co-authored-by: default avatarbudefei <budefei@cambricon.com>
      Co-authored-by: default avatarbudefei <budefei@cambricom.com>
      362a90f8
  16. 14 Apr, 2022 1 commit
  17. 21 Mar, 2022 1 commit
  18. 01 Mar, 2022 1 commit
  19. 09 Feb, 2022 1 commit
  20. 10 Jan, 2022 1 commit
  21. 29 Dec, 2021 1 commit
  22. 23 Dec, 2021 1 commit
  23. 10 Dec, 2021 1 commit
  24. 01 Dec, 2021 1 commit
  25. 28 Nov, 2021 1 commit
  26. 23 Nov, 2021 1 commit
    • q.yao's avatar
      Refactor csrc with device dispatcher (#1463) · 230f9a3b
      q.yao authored
      * Add device registry for pytorch ops
      
      * add declaration of CheckDeviceConsistency
      
      * fix for torch130
      
      * assert with torch check
      
      * Refactor ops with dispatch
      
      * update rest ops
      
      * faster install
      
      * update compatibility
      
      * update compatibility, rename parameter
      
      * move cpu implement to pytorch/cpu
      
      * update ops/csrc/README.md
      
      * fix rocm support
      
      * update cn document
      
      * update docs
      
      * list instead of map
      230f9a3b
  27. 01 Sep, 2021 1 commit
  28. 27 Aug, 2021 1 commit
  29. 18 Aug, 2021 1 commit
  30. 10 Aug, 2021 2 commits
    • Haodong Duan's avatar
      [Fix] Do not depend on setuptools>=52 (#1235) · 93418560
      Haodong Duan authored
      * use packaging instead
      
      * update
      
      * update
      
      * update
      
      * update
      93418560
    • Zaida Zhou's avatar
      [Refactor] Refactor the directory of csrc (#1206) · dfb48c87
      Zaida Zhou authored
      
      
      * [Refactor] Refactor the csrc directory
      
      * update MANIFEST.in
      
      * fix hip
      
      * add csrc readme
      
      * trailing whitespace
      
      * fix syntax error in setup.py
      
      * add compatibility docs
      
      * move parrots_cudawarpfunction.cuh to common/cuda
      
      * fix grammar, update directory tree
      
      * fix MANIFEST.in
      
      * Add new structre of csrc in compatibility.md
      
      * Add original structre of csrc in compatibility.md
      
      * fix typo
      
      * remove TODO
      
      * modify according to comment
      
      * format
      Co-authored-by: default avatargrimoire <yaoqian@sensetime.com>
      dfb48c87
  31. 09 Jul, 2021 1 commit
    • Y_Xuan's avatar
      [Feature] Porting mmcv for hip (#1022) · 2dc0a219
      Y_Xuan authored
      * porting mmcv for hip
      
      * add nvcc
      
      * fix format
      
      * fix format
      
      * fix bug for carafe
      
      * fix test_utils because rocm_torch not allow set torch.backends.cudnn.benchmark to false
      
      * add LOOSEVERSION
      
      * fix format
      
      * fix format of version
      
      * fix code format
      
      * test for yaml
      
      * fix bug for citest
      
      * fix bug for how to get torch._version_ at setup.py
      2dc0a219
  32. 17 Apr, 2021 1 commit
  33. 26 Feb, 2021 1 commit
    • z55250825's avatar
      Add new parrots extension implementation for all ops (#794) · 48d99025
      z55250825 authored
      * delete all parrots file
      add bbox_overlaps new parrots op impl
      
      * support first new impl parrts op (bbox_overlaps)(success test)
      
      * add box_iou_rotated op, test succeed
      
      * add carafe and carafe_naive op, test succeed (one parrots bug need fix)
      
      * add cc_attention op, test success
      
      * add corner_pool op, test success
      
      * add parrots op deform_conv, test success
      
      * add deform_roi_pool op, test success (but has question)
      
      * add focal loss op, test success (gradcheck)
      
      * add masked_conv2d op, test success
      
      * add modulated_deform_conv op, test success
      
      * add nms and nms_rotated op, test success
      
      * add psamask op, test success
      
      * add roi_align op, test_success
      
      * add roi_pool op, test success
      
      * add sync_bn op, test success
      
      * add tin_shift op, test success
      
      * fix test_deform_roi_pool, add parrots test
      
      * skip test_onnx because parrots does not support onnx
      
      * fix c++ lint
      
      * fix python lint
      
      * fix python lint
      48d99025
  34. 06 Jan, 2021 1 commit
    • q.yao's avatar
      [Feature]: Add custom operators support for TensorRT in mmcv (#686) · 0de9e149
      q.yao authored
      
      
      * start trt plugin prototype
      
      * Add test module, modify roialign convertor
      
      * finish roi_align trt plugin
      
      * fix conflict of RoiAlign and MMCVRoiAlign
      
      * fix for lint
      
      * fix test tensorrt module
      
      * test_tensorrt move import to test func
      
      * add except error type
      
      * add tensorrt to setup.cfg
      
      * code format with yapf
      
      * fix for clang-format
      
      * move tensorrt_utils to mmcv/tensorrt, add comments, better test module
      
      * fix line endings, docformatter
      
      * isort init, remove trailing whitespace
      
      * add except type
      
      * fix setup.py
      
      * put import extension inside trt setup
      
      * change c++ guard, update pytest script, better setup, etc
      
      * sort import with isort
      
      * sort import with isort
      
      * move init of plugin lib to init_plugins.py
      
      * resolve format and add test dependency: tensorrt
      
      * tensorrt should be installed from source not from pypi
      
      * update naming style and input check
      
      * resolve lint error
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      0de9e149
  35. 23 Dec, 2020 1 commit
    • q.yao's avatar
      [Feature]: Add custom operators support for onnxruntime in mmcv (#612) · 94810f22
      q.yao authored
      
      
      * add onnx support to roi_align and roi_pool
      
      * add softnms ort support
      
      * fix for lint
      
      * format cpp code with clang-format:google
      
      * add new empty line to the end of head files in onnxruntime
      
      * update to pytorch1.7
      
      * add test of softnms to onnxruntime
      
      * fix for lint
      
      * remote print in ops/info.py
      
      * change import order, fix for flake8
      
      * fix include
      
      * add assert torch>=1.7.0
      
      * [doc]: add document for onnxruntime custom operator
      
      * update onnxruntime version to v1.5.1 for softnms
      
      * remove doc menu
      
      * Resolve lint for markdown
      
      * resolve naming style in onnxruntime_op.md
      
      * Use old cpp apis, optimize test_onnx.py
      
      * Fixing strings in tests/test_ops/test_onnx.py
      
      * code format with yapf
      
      * fix soft_nms parrot
      
      * add import in onnxruntime setup, avoid conflict
      
      * fix doc and add assert
      
      * change cpp guard
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      94810f22