diff options
Diffstat (limited to 'test/test.cc')
-rw-r--r-- | test/test.cc | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test/test.cc b/test/test.cc index 3559738..45c27ad 100644 --- a/test/test.cc +++ b/test/test.cc @@ -1,6 +1,8 @@ #define CATCH_CONFIG_RUNNER #include "test.h" +#include <cmath> + int main(int argc, char ** argv) { return Catch::Session().run(argc, argv); } @@ -13,13 +15,13 @@ void CompareMSE(const float *float_ref, const float *int_ref, const float *int_t for (std::size_t i = 0; i < size; ++i) { float int_diff = int_ref[i] - int_test[i]; float float_diff = float_ref[i] - int_test[i]; - CHECK_MESSAGE(fabs(int_diff) <= int_tolerance, test_info << "Inaccurate compared to int reference at " << i << ' ' << int_ref[i] << ' ' << int_test[i]); - CHECK_MESSAGE(fabs(float_diff) <= float_tolerance, test_info << "Inaccurate compared to float reference at " << i << ' ' << float_ref[i] << ' ' << int_test[i]); + CHECK_MESSAGE(std::fabs(int_diff) <= int_tolerance, test_info << "Inaccurate compared to int reference at " << i << ' ' << int_ref[i] << ' ' << int_test[i]); + CHECK_MESSAGE(std::fabs(float_diff) <= float_tolerance, test_info << "Inaccurate compared to float reference at " << i << ' ' << float_ref[i] << ' ' << int_test[i]); int_sum += int_diff * int_diff; float_sum += float_diff * float_diff; } - CHECK_MESSAGE(fabs(sqrt(float_sum / size)) <= MSE_float_tolerance, test_info << "Float MSE = " << sqrt(float_sum / size)); - CHECK_MESSAGE(fabs(sqrt(int_sum / size)) <= MSE_int_tolerance, test_info << "Int MSE = " << sqrt(int_sum / size)); + CHECK_MESSAGE(std::fabs(sqrt(float_sum / size)) <= MSE_float_tolerance, test_info << "Float MSE = " << sqrt(float_sum / size)); + CHECK_MESSAGE(std::fabs(sqrt(int_sum / size)) <= MSE_int_tolerance, test_info << "Int MSE = " << sqrt(int_sum / size)); } } // namespace intgemm |