mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
b2fd224f27
Summary: Follow https://github.com/pytorch/pytorch/pull/123848, and test more c10 util functions. Pull Request resolved: https://github.com/pytorch/pytorch/pull/123900 Approved by: https://github.com/chenyang78
24 lines
605 B
C++
24 lines
605 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <ATen/NumericUtils.h>
|
|
#include <c10/util/generic_math.h>
|
|
#include <cmath>
|
|
namespace torch {
|
|
namespace aot_inductor {
|
|
|
|
TEST(TestMath, TestDivFloor) {
|
|
EXPECT_EQ(c10::div_floor_floating(5., 0.), INFINITY);
|
|
EXPECT_DOUBLE_EQ(c10::div_floor_floating(5., 2.), 2.);
|
|
EXPECT_DOUBLE_EQ(c10::div_floor_floating(5., -2.), -3.);
|
|
EXPECT_EQ(c10::div_floor_integer(5, 2), 2);
|
|
EXPECT_EQ(c10::div_floor_integer(5, -2), -3);
|
|
}
|
|
|
|
TEST(TestMath, TestNan) {
|
|
EXPECT_FALSE(at::_isnan(1.0));
|
|
EXPECT_TRUE(at::_isnan(std::nan("")));
|
|
}
|
|
|
|
} // namespace aot_inductor
|
|
} // namespace torch
|