• W
    fix bug for crmm model test=develop (#2786) · af95a0bb
    Wilber 提交于
    - modify aligned_matmul kernel for dynamically malloc memory
    - fix top_k_avg_pooling kernel to support data whose size is more than cuda shared memory.
    af95a0bb
sequence_topk_avg_pooling_compute.h 1.3 KB