diff options
Diffstat (limited to 'compiler/one-cmds/tests/pytorch-operations/state_dict_model.test')
-rw-r--r-- | compiler/one-cmds/tests/pytorch-operations/state_dict_model.test | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/compiler/one-cmds/tests/pytorch-operations/state_dict_model.test b/compiler/one-cmds/tests/pytorch-operations/state_dict_model.test new file mode 100644 index 000000000..ecd2a8112 --- /dev/null +++ b/compiler/one-cmds/tests/pytorch-operations/state_dict_model.test @@ -0,0 +1,39 @@ +#!/bin/bash + +# Copyright (c) 2021 Samsung Electronics Co., Ltd. All Rights Reserved +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +# Test one-import-pytorch ability to import NN model from .py file and serialized state_dict file. + +filename_ext="$(basename -- $0)" +filename="${filename_ext%.*}" + +trap_err_onexit() +{ + echo "${filename_ext} FAILED" + exit 255 +} + +trap trap_err_onexit ERR + +outputfile="state_dict_model.circle" + +# run test +one-import-pytorch --input_path=state_dict_model.pth --python_path=aux_generator.py --output_path=${outputfile} --input_shapes=1,2,3,3 --input_types=float32 &> /dev/null + +if [[ ! -s "${outputfile}" ]]; then + trap_err_onexit +fi + +echo "${filename_ext} SUCCESS" |