You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Leo Chen 2d95280e1f Feature/Enable Auto-Mixed-Precision in dynamic graph (#24903) ... * add auto_cast, test=develop * add loss scaler, test=develop * add comments, test=develop * refine code, test=develop * refine code, test=develop * do not set flags automatically, test=develop * fix custom op bug, test=develop * add more test, test=develop * refine enable logic, test=develop * enable amp test with GPU, test=develop * add unittest * add test for found_inf * follow comments * follow comments * remove global variable, use singleton * add some notes * update comments * update comments * update comments * add use_dynamic_loss_scaling argument * refine found_inf * refine found_inf		5 years ago
..
CMakeLists.txt	…
amp_check_finite_and_scale_op.cc	Feature/Enable Auto-Mixed-Precision in dynamic graph (#24903)	5 years ago
amp_check_finite_and_scale_op.cu	Feature/Enable Auto-Mixed-Precision in dynamic graph (#24903)	5 years ago
amp_check_finite_and_scale_op.h	…