• H
    Support memory eager deletion on recurrent OP (#17710) · 89bc3fd8
    Huihuang Zheng 提交于
    Test PaddingRNN on V100 GPU device.
    
    Test configuration: large model, padding mode (which is the mode using recurrentOp), one GPU.
                       
    GPU memory (MiB):   6414 (this PR)     vs   6837 (without this PR)
    Speed (steps/s):         10.28 (this PR)    vs    9.89 (without this PR)
     
    89bc3fd8
rnn_memory_helper_op.cc 6.3 KB