• anj-s's avatar
    [refactor] Refactor and enable multiprocess nn.Pipe benchmarks. (#319) · cd186441
    anj-s authored
    
    
    * mp cleanup
    
    * round of multiprocess refactoring
    
    * test golden run
    
    * print cuda stats
    
    * fix lint errors
    
    * enable multiprocess pipe benchmarks
    
    * set world size to be available gpus
    
    * more changes
    
    * use synthetic loaders for intermediate pipeline stages
    
    * merged master
    
    * fix for the devices property
    
    * dataloader fix
    
    * modify rank check
    
    * print wps stats
    
    * enable verification
    
    * fix logging
    
    * fix flag name
    
    * fix flag name
    
    * check for rank
    
    * fix indent
    
    * pass args
    
    * pass args
    
    * modify golden data
    
    * remove unused print messsage
    
    * fix lint errors
    
    * add comments
    
    * fix benchmarks
    Co-authored-by: default avatarAnjali Sridhar <anj@devfair0443.h2.fair>
    cd186441
wikitext2_data.py 2.92 KB