• Y
    [AMP] Reimplement check_nan_inf as check_numerics_kernel. (#52245) · 44bd5927
    Yiqun Liu 提交于
    * Reimplement the check_nan_inf function as check_numerics kernel.
    
    * Remove the cpu implemention to phi.
    
    * Add ifdef for the including of omp.h.
    
    * Move the use of FLAGS_check_nan_inf_level out of header file.
    
    * Implement a common PrintAndThrowError function.
    
    * Fix the error using of __NVCC__, which should be instead with __CUDA_ARCH__.
    
    * Add dependency of phi.
    
    * Polish codes and unittest.
    44bd5927
test_tensor_checker.py 4.3 KB