• Z
    Add flags to limit gpu memory (#22793) · d41d802b
    Zeng Jinle 提交于
    * add recorded cuda memory apis, fix typo, test=develop
    
    * add more ut, test=develop
    
    * follow comments, test=develop
    
    * fix py35 incompatible issues, test=develop
    d41d802b
test_limit_gpu_memory.cu 2.7 KB