Support compilation via Torchdynamo, AOT Autograd, NVFuser #17308
Support compilation via Torchdynamo, AOT Autograd, NVFuser
23962182
stas00
requested changes
on 2022-05-17
Address comments
0b4c279a
sgugger
approved these changes
on 2022-05-19
Lint
52503f2f
stas00
requested changes
on 2022-05-19
Stas comments - missing quality test
298b3baa
Lintere
6756fa53
Quality test
28f80ec0
stas00
commented
on 2022-05-19
Doc lint
bf417042
Reset CUDA peak mem
7925fa33
Add CustomTrainer
ce476195
require a single gpu
9daaf930
stas00
approved these changes
on 2022-05-25
sgugger
merged
897a8dd8
into main 3 years ago
Login to write a write a comment.
Login via GitHub