1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#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
|
