diff --git a/src/utilities/error.cu b/src/utilities/error.cu index e34fad802..296474980 100644 --- a/src/utilities/error.cu +++ b/src/utilities/error.cu @@ -14,6 +14,7 @@ */ #include "error.cuh" +#include #include #include #include @@ -176,6 +177,14 @@ float get_float_from_token(const std::string& token, const char* filename, const std::cout << " Error message: " << e.what() << std::endl; exit(1); } + if (std::isinf(value)) { + std::cout << "This number is inf.\n"; + exit(1); + } + if (std::isnan(value)) { + std::cout << "This number is nan.\n"; + exit(1); + } return value; } @@ -191,5 +200,13 @@ double get_double_from_token(const std::string& token, const char* filename, con std::cout << " Error message: " << e.what() << std::endl; exit(1); } + if (std::isinf(value)) { + std::cout << "This number is inf.\n"; + exit(1); + } + if (std::isnan(value)) { + std::cout << "This number is nan.\n"; + exit(1); + } return value; }