Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[TOPI][OP] Use Thrust sort for argsort and topk #5097

Merged
merged 10 commits into from
Mar 20, 2020
Merged

[TOPI][OP] Use Thrust sort for argsort and topk #5097

merged 10 commits into from
Mar 20, 2020

Commits on Mar 19, 2020

  1. [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.
    @kazum
    kazum committed Mar 19, 2020
    Configuration menu
    Copy the full SHA
    a2d48ee View commit details
    Browse the repository at this point in the history

  2. cmake: make CUDA optional

    @kazum
    kazum committed Mar 19, 2020
    Configuration menu
    Copy the full SHA
    4823e93 View commit details
    Browse the repository at this point in the history

  3. allow .cu file to be into the repository

    @kazum
    kazum committed Mar 19, 2020
    Configuration menu
    Copy the full SHA
    98b2aa3 View commit details
    Browse the repository at this point in the history

  4. pylint fix and cleanup

    @kazum
    kazum committed Mar 19, 2020
    Configuration menu
    Copy the full SHA
    13b3edc View commit details
    Browse the repository at this point in the history

  5. require cmake 3.8 only when thrust is enabled

    @kazum
    kazum committed Mar 19, 2020
    Configuration menu
    Copy the full SHA
    addba76 View commit details
    Browse the repository at this point in the history

  6. fix nvcc compiler error when passing -pthread

    @kazum
    kazum committed Mar 19, 2020
    Configuration menu
    Copy the full SHA
    041860e View commit details
    Browse the repository at this point in the history

  7. add missing include

    @kazum
    kazum committed Mar 19, 2020
    Configuration menu
    Copy the full SHA
    0d88876 View commit details
    Browse the repository at this point in the history

Commits on Mar 20, 2020

  1. add USE_THRUST option in config.cmake

    @kazum
    kazum committed Mar 20, 2020
    Configuration menu
    Copy the full SHA
    b1e388d View commit details
    Browse the repository at this point in the history

  2. retrigger CI

    @kazum
    kazum committed Mar 20, 2020
    Configuration menu
    Copy the full SHA
    bc06ce4 View commit details
    Browse the repository at this point in the history

  3. retrigger CI

    @kazum
    kazum committed Mar 20, 2020
    Configuration menu
    Copy the full SHA
    cc59cbf View commit details
    Browse the repository at this point in the history