mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
1216e5fe7f
* [Docs] Use PyTorch sphinx theme * add sphinx-copybutton * add twitter and zhihu link * remove shpinx_rtd_theme from doc.xtx * update docs.txt * update conf.py
104 lines
3.1 KiB
Markdown
104 lines
3.1 KiB
Markdown
### v1.3.11
|
||
|
||
In order to flexibly support more backends and hardwares like `NVIDIA GPUs` and `AMD GPUs`, the directory of `mmcv/ops/csrc` is refactored. Note that this refactoring will not affect the usage in API. For related information, please refer to [PR1206](https://github.com/open-mmlab/mmcv/pull/1206).
|
||
|
||
The original directory was organized as follows.
|
||
|
||
```
|
||
.
|
||
├── common_cuda_helper.hpp
|
||
├── ops_cuda_kernel.cuh
|
||
├── pytorch_cpp_helper.hpp
|
||
├── pytorch_cuda_helper.hpp
|
||
├── parrots_cpp_helper.hpp
|
||
├── parrots_cuda_helper.hpp
|
||
├── parrots_cudawarpfunction.cuh
|
||
├── onnxruntime
|
||
│ ├── onnxruntime_register.h
|
||
│ ├── onnxruntime_session_options_config_keys.h
|
||
│ ├── ort_mmcv_utils.h
|
||
│ ├── ...
|
||
│ ├── onnx_ops.h
|
||
│ └── cpu
|
||
│ ├── onnxruntime_register.cpp
|
||
│ ├── ...
|
||
│ └── onnx_ops_impl.cpp
|
||
├── parrots
|
||
│ ├── ...
|
||
│ ├── ops.cpp
|
||
│ ├── ops_cuda.cu
|
||
│ ├── ops_parrots.cpp
|
||
│ └── ops_pytorch.h
|
||
├── pytorch
|
||
│ ├── ...
|
||
│ ├── ops.cpp
|
||
│ ├── ops_cuda.cu
|
||
│ ├── pybind.cpp
|
||
└── tensorrt
|
||
├── trt_cuda_helper.cuh
|
||
├── trt_plugin_helper.hpp
|
||
├── trt_plugin.hpp
|
||
├── trt_serialize.hpp
|
||
├── ...
|
||
├── trt_ops.hpp
|
||
└── plugins
|
||
├── trt_cuda_helper.cu
|
||
├── trt_plugin.cpp
|
||
├── ...
|
||
├── trt_ops.cpp
|
||
└── trt_ops_kernel.cu
|
||
```
|
||
|
||
After refactored, it is organized as follows.
|
||
|
||
```
|
||
.
|
||
├── common
|
||
│ ├── box_iou_rotated_utils.hpp
|
||
│ ├── parrots_cpp_helper.hpp
|
||
│ ├── parrots_cuda_helper.hpp
|
||
│ ├── pytorch_cpp_helper.hpp
|
||
│ ├── pytorch_cuda_helper.hpp
|
||
│ └── cuda
|
||
│ ├── common_cuda_helper.hpp
|
||
│ ├── parrots_cudawarpfunction.cuh
|
||
│ ├── ...
|
||
│ └── ops_cuda_kernel.cuh
|
||
├── onnxruntime
|
||
│ ├── onnxruntime_register.h
|
||
│ ├── onnxruntime_session_options_config_keys.h
|
||
│ ├── ort_mmcv_utils.h
|
||
│ ├── ...
|
||
│ ├── onnx_ops.h
|
||
│ └── cpu
|
||
│ ├── onnxruntime_register.cpp
|
||
│ ├── ...
|
||
│ └── onnx_ops_impl.cpp
|
||
├── parrots
|
||
│ ├── ...
|
||
│ ├── ops.cpp
|
||
│ ├── ops_parrots.cpp
|
||
│ └── ops_pytorch.h
|
||
├── pytorch
|
||
│ ├── info.cpp
|
||
│ ├── pybind.cpp
|
||
│ ├── ...
|
||
│ ├── ops.cpp
|
||
│ └── cuda
|
||
│ ├── ...
|
||
│ └── ops_cuda.cu
|
||
└── tensorrt
|
||
├── trt_cuda_helper.cuh
|
||
├── trt_plugin_helper.hpp
|
||
├── trt_plugin.hpp
|
||
├── trt_serialize.hpp
|
||
├── ...
|
||
├── trt_ops.hpp
|
||
└── plugins
|
||
├── trt_cuda_helper.cu
|
||
├── trt_plugin.cpp
|
||
├── ...
|
||
├── trt_ops.cpp
|
||
└── trt_ops_kernel.cu
|
||
```
|