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
26 lines
587 B
C++
26 lines
587 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <c10/util/TypeCast.h>
|
|
#include <c10/util/bit_cast.h>
|
|
namespace torch {
|
|
namespace aot_inductor {
|
|
|
|
TEST(TestCast, TestConvert) {
|
|
c10::BFloat16 a = 3.0f;
|
|
c10::Half b = 3.0f;
|
|
|
|
EXPECT_EQ(c10::convert<c10::Half>(a), b);
|
|
EXPECT_EQ(a, c10::convert<c10::BFloat16>(b));
|
|
}
|
|
|
|
TEST(TestCast, TestBitcast) {
|
|
c10::BFloat16 a = 3.0f;
|
|
c10::Half b = 3.0f;
|
|
|
|
EXPECT_EQ(c10::bit_cast<c10::BFloat16>(c10::bit_cast<c10::Half>(a)), a);
|
|
EXPECT_EQ(c10::bit_cast<c10::Half>(c10::bit_cast<c10::BFloat16>(b)), b);
|
|
}
|
|
|
|
} // namespace aot_inductor
|
|
} // namespace torch
|