1. 23 Mar, 2020 1 commit
    • Add thrust support for nms (#5116) · 25d35421
      * add argsort_nms_thrust
      
      * consider valid count in thrust nms sort
      
      * make thrust optional
      
      * typo
      
      * typo
      
      * fix pylint
      
      * address some of the comments
      
      * address more comments
      
      * fix lint
      
      * address more comments
      
      * address more comments
      Leyuan Wang committed
  2. 20 Mar, 2020 1 commit
    • [TOPI][OP] Use Thrust sort for argsort and topk (#5097) · 2e8f3a91
      * [TOPI][OP] Use Thrust sort for argsort and topk
      
      The current GPU sort implementation (odd-even transposition sort) is too slow
      when the number of elements is large.  This PR introduces Thrust implementation
      of sort which is much faster.
      
      Note that this change requires CMake 3.8 or later since we have to use nvcc to
      compile a thrust code.
      
      * cmake: make CUDA optional
      
      * allow .cu file to be into the repository
      
      * pylint fix and cleanup
      
      * require cmake 3.8 only when thrust is enabled
      
      * fix nvcc compiler error when passing -pthread
      
      * add missing include
      
      * add USE_THRUST option in config.cmake
      
      * retrigger CI
      
      * retrigger CI
      MORITA Kazutaka committed