• 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
gpu_external_function.cpp 29.9 KB