• Anthony Chen's avatar
    fix attribute mismatch for memory profiler · 99c65490
    Anthony Chen authored
    Summary:
    Pull Request resolved: https://github.com/facebookresearch/d2go/pull/544
    
    The previous diff on memory profiler D45673764 doesn't pick up a config key name change and causes an attribute not found error. This diff fixes it and adds two unittests (one with gpu one without) for using memory profiler in runner
    
    Reviewed By: wat3rBro
    
    Differential Revision: D46114730
    
    fbshipit-source-id: d066d435021983d90f4a75e0c88798a3aedcaf92
    99c65490
test_gpu_memory_profiler.py 2.98 KB