• Y
    Optimize the concat and split kernel for specical cases when the number of... · 5782ddda
    Yiqun Liu 提交于
    Optimize the concat and split kernel for specical cases when the number of inputs/outputs is 2 (#17415)
    
    * Optimize the concat and split kernel for special cases that the number of inputs/outputs is 2.
    test=develop
    
    * Refine codes.
    test=develop
    
    * Correct the condition.
    test=develop
    
    * Move the define of tmp_data outside the if statement.
    
    * Print the cudnn minor version.
    test=develop
    
    * Fix the case when in_num/o_num is 1 in concat/split op.
    test=develop
    
    * Remove const_cast.
    test=develop
    5782ddda
cudnn.cmake 3.2 KB