52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
#include <gtest/gtest.h>
|
|
#include <torch/nativert/graph/Serialization.h>
|
|
|
|
namespace torch::nativert {
|
|
TEST(SerializationTest, CheckIsSymbolic) {
|
|
torch::_export::TensorArgument tensor_arg;
|
|
torch::_export::Argument as_tensor_arg;
|
|
as_tensor_arg.set_as_tensor(tensor_arg);
|
|
EXPECT_TRUE(isSymbolic(as_tensor_arg));
|
|
|
|
std::vector<torch::_export::TensorArgument> tensor_args;
|
|
torch::_export::Argument as_tensors_arg;
|
|
as_tensors_arg.set_as_tensors(tensor_args);
|
|
EXPECT_TRUE(isSymbolic(as_tensors_arg));
|
|
|
|
torch::_export::SymIntArgument sym_int_arg;
|
|
torch::_export::Argument as_sym_int_arg;
|
|
as_sym_int_arg.set_as_sym_int(sym_int_arg);
|
|
EXPECT_TRUE(isSymbolic(as_sym_int_arg));
|
|
|
|
torch::_export::Argument as_int_arg;
|
|
as_int_arg.set_as_int(static_cast<int64_t>(1));
|
|
EXPECT_FALSE(isSymbolic(as_int_arg));
|
|
|
|
torch::_export::Argument as_bool_arg;
|
|
as_bool_arg.set_as_bool(true);
|
|
EXPECT_FALSE(isSymbolic(as_bool_arg));
|
|
|
|
torch::_export::Argument as_string_arg;
|
|
as_string_arg.set_as_string("test_string");
|
|
EXPECT_FALSE(isSymbolic(as_string_arg));
|
|
}
|
|
|
|
TEST(SerializationTest, ConstantToValue) {
|
|
torch::_export::Argument as_int_arg;
|
|
as_int_arg.set_as_int(static_cast<int64_t>(42));
|
|
auto value = constantToValue(as_int_arg, false);
|
|
EXPECT_EQ(value, Constant(static_cast<int64_t>(42)));
|
|
|
|
torch::_export::Argument as_bool_arg;
|
|
as_bool_arg.set_as_bool(true);
|
|
value = constantToValue(as_bool_arg, false);
|
|
EXPECT_EQ(value, Constant(true));
|
|
|
|
torch::_export::Argument as_string_arg;
|
|
as_string_arg.set_as_string("test_string");
|
|
value = constantToValue(as_string_arg, false);
|
|
EXPECT_EQ(value, Constant("test_string"));
|
|
}
|
|
|
|
} // namespace torch::nativert
|