wehub-resource-sync
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
# Copyright (c) 2020 PaddlePaddle Authors. 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.
|
|
|
|
import inspect
|
|
import textwrap
|
|
import unittest
|
|
|
|
import numpy as np
|
|
from dygraph_to_static_utils import (
|
|
Dy2StTestBase,
|
|
static_guard,
|
|
test_ast_only,
|
|
)
|
|
from ifelse_simple_func import (
|
|
dyfunc_with_if_else,
|
|
dyfunc_with_if_else2,
|
|
nested_if_else,
|
|
)
|
|
|
|
import paddle
|
|
import paddle.nn.functional as F
|
|
from paddle import base
|
|
from paddle.jit.dy2static.utils import ast_to_func
|
|
from paddle.utils import gast
|
|
|
|
|
|
class TestAST2Func(Dy2StTestBase):
|
|
"""
|
|
TestCase for the transformation from ast.AST into python callable function.
|
|
"""
|
|
|
|
def _ast2func(self, func):
|
|
source = inspect.getsource(func)
|
|
source = textwrap.dedent(source)
|
|
ast_root = gast.parse(source)
|
|
transformed_func, _ = ast_to_func(ast_root, func)
|
|
return transformed_func
|
|
|
|
@test_ast_only
|
|
def test_ast2func(self):
|
|
def func(x, y):
|
|
return x + y
|
|
|
|
x, y = 10, 20
|
|
self.assertEqual(func(x, y), self._ast2func(func)(x, y))
|
|
|
|
@test_ast_only
|
|
def test_ast2func_dygraph(self):
|
|
funcs = [dyfunc_with_if_else, dyfunc_with_if_else2, nested_if_else]
|
|
x_data = np.random.random([10, 16]).astype('float32')
|
|
for func in funcs:
|
|
x_v = paddle.to_tensor(x_data)
|
|
true_ret = func(x_v).numpy()
|
|
test_ret = self._ast2func(func)(x_v).numpy()
|
|
np.testing.assert_allclose(true_ret, test_ret)
|
|
|
|
@test_ast_only
|
|
def test_ast2func_static(self):
|
|
def func(x):
|
|
y = F.relu(x)
|
|
loss = paddle.mean(y)
|
|
return loss
|
|
|
|
x_data = np.random.random([10, 16]).astype('float32')
|
|
with static_guard():
|
|
main_program = paddle.static.Program()
|
|
with base.program_guard(main_program):
|
|
x_v = paddle.assign(x_data)
|
|
true_ret = func(x_v)
|
|
test_ret = self._ast2func(func)(x_v)
|
|
exe = paddle.static.Executor(paddle.CPUPlace())
|
|
ret = exe.run(main_program, fetch_list=[true_ret, test_ret])
|
|
np.testing.assert_allclose(ret[0], ret[1])
|
|
|
|
@test_ast_only
|
|
def test_ast2func_error(self):
|
|
with self.assertRaises(Exception) as e:
|
|
self.assertRaises(TypeError, ast_to_func("x = a + b", 'foo'))
|
|
self.assertIn("'foo' is not a Python function", str(e.exception))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|