• Y
    Enhance check_nan_inf implementation for CPU. (#48591) · 69e695b7
    Yiqun Liu 提交于
    * Enable to print device info.
    
    * Enhance the nan and inf checking for cpu.
    
    * Implement a common print function.
    
    * Unify the check of complex numbers.
    
    * Rewrite the omp method.
    
    * Count and print the number of nan and inf.
    
    * Change the print content.
    
    * Add unittest.
    69e695b7
nan_inf_utils_detail.cu 17.5 KB