mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
eba5fc91ac
Summary: Serialization contains utilities to deserialize a graph saved on disk in json format as defined in `torch/csrc/utils/generated_serialization_types.h` to the in-memory representation as defined in `torch/nativert/graph/Graph.h` Test Plan: buck2 run @mode/dev-nosan caffe2/test/cpp/nativert:serialization_test Rollback Plan: Differential Revision: D76012641 Pull Request resolved: https://github.com/pytorch/pytorch/pull/155229 Approved by: https://github.com/zhxchen17
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
|