• Adam Becker's avatar
    mixed changes · c89cab31
    Adam Becker 提交于
     - add float16 suppport
     - fix 64 bit input bug
     - fix int8 bug due to integer overflow
     - more checking on k
     - add license from pytorch
    c89cab31
k_topk_common.cuh 8.3 KB