1. 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
  2. 27 Jan, 2020 1 commit
  3. 19 Jan, 2020 1 commit
  4. 27 Oct, 2019 1 commit
  5. 08 Apr, 2019 1 commit
    • [HEADER] Add Header to Comply with ASF Release Policy (#2982) · cffb4fba
      * [HEADER] ASF header dir=include
      
      * [HEADER] ASF Header dir=src
      
      * [HEADER] ASF Header -dir=python
      
      * [HEADER] ASF header dir=topi
      
      * [HEADER] ASF Header dir=nnvm
      
      * [HEADER] ASF Header -dir=tutorials
      
      * [HEADER] ASF Header dir=tests
      
      * [HEADER] ASF Header -dir=docker
      
      * fix whitespace
      
      * [HEADER] ASF Header -dir=jvm
      
      * [HEADER] ASF Header -dir=web
      
      * [HEADER] ASF Header --dir=apps
      
      * [HEADER] ASF Header --dir=vta
      
      * [HEADER] ASF Header -dir=go
      
      * temp
      
      * [HEADER] ASF Header --dir=rust
      
      * [HEADER] Add ASF Header --dir=cmake
      
      * [HEADER] ASF Header --dir=docs
      
      * [HEADER] Header for Jenkinsfile
      
      * [HEADER] ASF Header to toml and md
      
      * [HEADER] ASF Header to gradle
      
      * Finalize rat cleanup
      
      * Fix permission
      
      * Fix java test
      
      * temporary remove nnvm onnx test
      Tianqi Chen committed
  6. 13 Jun, 2018 1 commit
  7. 11 Jun, 2018 1 commit