| 28 | #include "gtest/gtest.h" |
| 29 | |
| 30 | TEST(DeviceTypeCompatTest, DeviceTypeConversionAndStreamOperator) { |
| 31 | EXPECT_EQ(c10::DeviceTypeToPhi(c10::DeviceType::CPU), |
| 32 | phi::AllocationType::CPU); |
| 33 | EXPECT_EQ(c10::DeviceTypeToPhi(c10::DeviceType::CUDA), |
| 34 | phi::AllocationType::GPU); |
| 35 | EXPECT_EQ(c10::DeviceTypeToPhi(c10::DeviceType::XPU), |
| 36 | phi::AllocationType::XPU); |
| 37 | EXPECT_EQ(c10::DeviceTypeToPhi(c10::DeviceType::IPU), |
| 38 | phi::AllocationType::IPU); |
| 39 | EXPECT_EQ(c10::DeviceTypeToPhi(c10::DeviceType::CUSTOM), |
| 40 | phi::AllocationType::CUSTOM); |
| 41 | EXPECT_EQ(c10::DeviceTypeToPhi(static_cast<c10::DeviceType>(-1)), |
| 42 | phi::AllocationType::UNDEFINED); |
| 43 | |
| 44 | EXPECT_EQ(c10::PhiToDeviceType(phi::AllocationType::CPU), |
| 45 | c10::DeviceType::CPU); |
| 46 | EXPECT_EQ(c10::PhiToDeviceType(phi::AllocationType::GPU), |
| 47 | c10::DeviceType::CUDA); |
| 48 | EXPECT_EQ(c10::PhiToDeviceType(phi::AllocationType::XPU), |
| 49 | c10::DeviceType::XPU); |
| 50 | EXPECT_EQ(c10::PhiToDeviceType(phi::AllocationType::IPU), |
| 51 | c10::DeviceType::IPU); |
| 52 | EXPECT_EQ(c10::PhiToDeviceType(phi::AllocationType::CUSTOM), |
| 53 | c10::DeviceType::CUSTOM); |
| 54 | EXPECT_EQ(c10::PhiToDeviceType(phi::AllocationType::UNDEFINED), |
| 55 | c10::DeviceType::CPU); |
| 56 | |
| 57 | EXPECT_TRUE(c10::isValidDeviceType(c10::DeviceType::CPU)); |
| 58 | EXPECT_TRUE(c10::isValidDeviceType(c10::DeviceType::CUSTOM)); |
| 59 | EXPECT_FALSE(c10::isValidDeviceType(static_cast<c10::DeviceType>(-9))); |
| 60 | |
| 61 | std::ostringstream cpu_os; |
| 62 | cpu_os << c10::DeviceType::CPU; |
| 63 | EXPECT_EQ(cpu_os.str(), "cpu"); |
| 64 | std::ostringstream cuda_os; |
| 65 | cuda_os << c10::DeviceType::CUDA; |
| 66 | EXPECT_EQ(cuda_os.str(), "cuda"); |
| 67 | std::ostringstream xpu_os; |
| 68 | xpu_os << c10::DeviceType::XPU; |
| 69 | EXPECT_EQ(xpu_os.str(), "xpu"); |
| 70 | std::ostringstream ipu_os; |
| 71 | ipu_os << c10::DeviceType::IPU; |
| 72 | EXPECT_EQ(ipu_os.str(), "ipu"); |
| 73 | std::ostringstream custom_os; |
| 74 | custom_os << c10::DeviceType::CUSTOM; |
| 75 | EXPECT_EQ(custom_os.str(), "privateuseone"); |
| 76 | std::ostringstream invalid_os; |
| 77 | invalid_os << static_cast<c10::DeviceType>(99); |
| 78 | EXPECT_TRUE(invalid_os.str().empty()); |
| 79 | |
| 80 | EXPECT_EQ(c10::DeviceType::PrivateUse1, c10::DeviceType::CUSTOM); |
| 81 | EXPECT_EQ(c10::kPrivateUse1, c10::DeviceType::PrivateUse1); |
| 82 | } |
| 83 | |
| 84 | TEST(DeviceCompatTest, DeviceParseAndPlaceBranches) { |
| 85 | c10::Device cpu("cpu"); |
nothing calls this directly
no test coverage detected