310 Commits

Author	SHA1	Message	Date
Jirka Borovec	5bce06c050	nb. devices (#2973)	2020-08-14 11:37:21 +02:00
William Falcon	0c264689cb	Fixes #2942 (#2969) ... * Fixes #2942 * doc fix	2020-08-13 21:54:57 -04:00
Jirka Borovec	4354690e55	add apex test (#2921) ... * add apex test * rename * level * events * wrap * evt * miss * apex * apex * apex * apex * apex * apex * Update tests/models/test_amp.py Co-authored-by: William Falcon <waf2107@columbia.edu> * notes * notes Co-authored-by: William Falcon <waf2107@columbia.edu>	2020-08-13 10:03:13 -04:00
Phil	e3528afae3	Move optimizer creation after device placement for ddp backends. (#2904)	2020-08-12 06:34:59 -04:00
Jirka Borovec	a6e7aa7796	allow using apex with any PT version (#2865) ... * wip * setup * type * name * wip * docs * imports * fix if * fix if * use_amp * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * fix tests * Apply suggestions from code review Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * fix tests * todos Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>	2020-08-08 11:07:32 +02:00
Jirka Borovec	b7d72706c3	clean imports (#2867) ... * clean imports * miss	2020-08-08 00:33:51 +02:00
Jirka Borovec	f8c058215f	simplify tests & cleaning (#2588) ... * simplify * tmpdir * revert * clean * accel * types * test * edit test acc Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> * Update test acc Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>	2020-08-07 23:22:05 +02:00
William Falcon	4dbd761a1c	refactor 3/n (#2709) ... * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator	2020-07-25 20:56:50 -04:00
William Falcon	b34217e410	Refactor 2/n (#2708) ... * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator	2020-07-25 17:31:34 -04:00
William Falcon	071e09fe38	refactor 1/n for v1.0.0 (#2704) ... * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator * reactor into gpu accelerator	2020-07-25 14:38:51 -04:00