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.

Zhen Wang 5ac71b36fb Remove tensor copy in the update_loss_scaling op. (#29426) ... * remove tensor copy in the update_loss_scaling op * not use thrust. * fix some cuda memory access error.		4 years ago
..
CMakeLists.txt	Compiling operator libraries with Unity build (#29130)	4 years ago
check_finite_and_unscale_op.cc	Update amp_check_finite_and_scale_op and add an updating_loss_scaling op for static graph amp training. (#26240)	5 years ago
check_finite_and_unscale_op.cu	optimizer amp, all use fp16 communication, overlap last comm and compute (#28957)	4 years ago
check_finite_and_unscale_op.h	Update amp_check_finite_and_scale_op and add an updating_loss_scaling op for static graph amp training. (#26240)	5 years ago
unity_build_rule.cmake	Compiling operator libraries with Unity build (#29130)	4 years ago
update_loss_scaling_op.cc	Remove tensor copy in the update_loss_scaling op. (#29426)	4 years ago
update_loss_scaling_op.cu	Remove tensor copy in the update_loss_scaling op. (#29426)	4 years ago
update_loss_scaling_op.h	Remove tensor copy in the update_loss_scaling op. (#29426)	4 years ago