• C
    Remove workspace_handle in conv_cudnn (#15186) · 064512aa
    chengduo 提交于
    * remove workspace_handle in conv2d_cudnn
    test=develop
    
    * remove workspace_handle
    test=develop
    
    * fix bug
    test=develop
    
    * make test_conv2d_op SERIAL
    test=develop
    
    * save memory in conv_cudnn
    test=develop
    
    * enhance thread safety
    test=develop
    
    * enhance temporary allocator
    test=develop
    
    * Add excess fraction
    test=develop
    
    * follow comments
    test=develop
    
    * fix bug and code refine
    test=develop
    
    * fix memory size check
    test=develop
    
    * rename reuse_tmp_allocation_excess_fraction
    test=develop
    064512aa
conv_cudnn_op.cu.cc 24.6 KB