Commit Graph

5 Commits (11e9876c3f33a5f710e766857385304adc1586a2)

Author SHA1 Message Date
lvhan028 6fdf6b8616
apply cmake-format and refactor cmake scripts of building onnx and te… (#99)
* apply cmake-format and refactor cmake scripts of building onnx and tensorrt ops

* add static target

* suppress 'CMAKE_CUDA_ARCHITECTURES' warning when cmake with version 18 or later is used

* fix typo
2021-10-13 16:42:11 +08:00
AllentDan 8fe8056080
[Feature] Add trt ort mdcn plugins (#43)
* add trt mdcn plugin

* add onnxruntime mdcn op

* add mdcn trt ort plugins

* fix lint

* remove comment

* remove plugin condition lines

* apply new form

* use serialized_size
2021-08-25 10:17:21 +08:00
q.yao 04a8046864
[Feature] add register for ort custom op (#21)
* add register for ort custom op

* ort_mmcv_util->ort_util

* ortApi->kOrtApi
2021-07-23 17:16:07 +08:00
grimoire 6eb2e89016 update cmakelists 2021-06-21 10:34:48 +08:00
grimoire 93408c7cd8 add type annotation
add onnxruntime plugin support, add more custom ops

update function_rewriter, enable use origin function

add setup.cfg

update setup.cfg
2021-06-16 15:36:58 +08:00