Function TEST

test/cpp/compat/torch_library_test.cc:139–152

}

TEST(test_torch_library, TestLibraryOperators) {
  auto qualified_name = "example_library::mymuladd";
  auto* op = torch::OperatorRegistry::instance().find_operator(qualified_name);
  ASSERT_NE(op, nullptr);
  auto impl_it = op->implementations.find(c10::DispatchKey::CPU);
  ASSERT_NE(impl_it, op->implementations.end());
  torch::FunctionArgs function_args;
  function_args.add_arg(torch::IValue(at::ones({2, 2}, at::kFloat)));
  function_args.add_arg(torch::IValue(at::ones({2, 2}, at::kFloat)));
  function_args.add_arg(torch::IValue(2.0));
  auto result = impl_it->second.call_with_args(function_args);
  ASSERT_TRUE(result.get_value().is_tensor());
  auto result_tensor = result.get_value().to_tensor();
}

TEST(test_torch_library, TestLibraryClasses) {
  auto qualified_name = "example_library::TestClass";