GitLab

Handle the use of .sum() in fused LSTM/GRU backward

amp shouldn't convert CPU tensors

DDP fix, imagenet speedup

Reverting setup.py to rely on cpp_extension's native CUDA_HOME, because upstream fixed cross-compilation

Read world_size argument in multiproc if specified in argslist

Print speed (img/sec) in imagenet example

This reverts commit 47ac5c2b.

Fix compatibility checks for 18.04 container

Fix regex expression to search for libaten

Create LARC.py