tensorplay.amp.autocast_mode.get_autocast_gpu_dtype

tensorplay.amp.autocast_mode.get_autocast_gpu_dtype()[source]

Return the dtype to be used for CUDA autocasting.

Warning

Kept for backward compatibility. Prefer get_autocast_dtype().