1. 22 5月, 2023 1 次提交
    • T
      Add multiclass_nms3 GPU kernel (#52401) · f71c805e
      Tian Zheng 提交于
      * Add GPU kernel for multiclass_nms3 op
      
      * Make multiclass_nms3 gpu kernel output consistent with cpu kernel
      
      * Fix API incompatibility
      
      * Fix unittests on builds without CUDA
      
      * Fix ROCM build
      
      * Remove fluid headers; Use default atol for unittest
      
      * Change function and variable naming
      
      * Add comments; Reduce redundant code
      
      * Use paddle test framework
      f71c805e