• Ayan Moitra's avatar
    Support TopK for NvidiaGPU backend (#1908) · d901446d
    Ayan Moitra authored
    * fresh commit for the changes
    
    * Working topk on ndims for nvGPU
    
    * fix
    
    * clang
    
    * Added unit test, improved kernel hash and Bob's comment
    
    * int64 test+clang
    
    * Moved argReduce and topk tests to a separate file
    
    * TopK unsupported for IntelGPU
    
    * addressed Fenglei and Chris's comments
    
    * addressed Fenglei and Chris's comments
    d901446d
Name
Last commit
Last update
.ci Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
.clang-format Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.gitmodules Loading commit data...
.travis.yml Loading commit data...
CMakeLists.txt Loading commit data...
CODEOWNERS Loading commit data...
CONTRIB.md Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
VERSION.in Loading commit data...
changes.md Loading commit data...