Function TEST

test/cpp/phi/core/test_kernel_factory.cc:31–41

// TODO(chenweihang): add more unittests later

TEST(KernelKey, ConstructAndOStream) {
  phi::KernelKey key(
      phi::Backend::CPU, phi::DataLayout::NCHW, phi::DataType::FLOAT32);
  EXPECT_EQ(key.backend(), phi::Backend::CPU);
  EXPECT_EQ(key.layout(), phi::DataLayout::NCHW);
  EXPECT_EQ(key.dtype(), phi::DataType::FLOAT32);
  std::ostringstream oss;
  oss << key;
  std::cout << oss.str();
  oss.flush();
}

TEST(KernelFactory, SelectedKernelMap) {
  auto kernel_map = phi::KernelFactory::Instance().SelectKernelMap("scale");