Searched refs:amp_update_scale_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
181 __global__ void amp_update_scale_cuda_kernel(float* current_scale, in amp_update_scale_cuda_kernel() function240 amp_update_scale_cuda_kernel<<<1, 1, 0, at::cuda::getCurrentCUDAStream()>>>( in _amp_update_scale_cuda_()