2021-09-14 16:35:41 +08:00
|
|
|
|
### v1.3.11
|
2021-08-10 15:09:19 +08:00
|
|
|
|
|
|
|
|
|
为了灵活地支持更多的后端和硬件,例如 `NVIDIA GPUs` 、`AMD GPUs`,我们重构了 `mmcv/ops/csrc` 目录。注意,这次重构不会影响 API 的使用。更多相关信息,请参考 [PR1206](https://github.com/open-mmlab/mmcv/pull/1206)。
|
|
|
|
|
|
|
|
|
|
原始的目录结构如下所示
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
.
|
|
|
|
|
├── 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
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
重构之后,它的结构如下所示
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
.
|
|
|
|
|
├── 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
|
|
|
|
|
```
|