• WilliamKyle's avatar
    [Feature] Add rotated_feature_align cpu & onnxruntime implementation (#1878) · b30755ee
    WilliamKyle authored
    * add rotated_feature_align cpu implementation
    
    * add rotated_feature_align onnxruntime implementation
    
    * Update code for advices from grimoire
    
    Remove useless comment from mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
    
    Replace ambiguous function name atomicAdd in mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
    
    Simplify unit test with parameter in tests/test_ops/test_rotated_feature_align.py
    
    Use fma in interpolate in mmcv/ops/csrc/onnxruntime/cpu/rotated_feature_align.cpp mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp
    
    * Inline the function to reduce the overhead of the function call
    
    Use fma in interpolate
    b30755ee
test_onnx.py 34.6 KB