未验证 提交 97b17cd8 编写于 作者: X xiaoguoguo626807 提交者: GitHub

【prim】set tanh_double comp api as default (#54302)

* set tanh_mutilply_double comp api as default

* modify timeout time

* rollback mutiply

* rollback mutiply
上级 099b3d25
...@@ -69,6 +69,7 @@ prim_white_list = [ ...@@ -69,6 +69,7 @@ prim_white_list = [
"subtract_double_grad", "subtract_double_grad",
"add_triple_grad", "add_triple_grad",
"silu_double_grad", "silu_double_grad",
"tanh_double_grad",
] ]
# dict of special api that forward api's output will affect bacward api's output # dict of special api that forward api's output will affect bacward api's output
......
...@@ -12,8 +12,10 @@ ...@@ -12,8 +12,10 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import sys
import unittest import unittest
sys.path.append("../../../legacy_test")
import gradient_checker import gradient_checker
import numpy as np import numpy as np
import parameterized as param import parameterized as param
...@@ -23,7 +25,7 @@ import paddle ...@@ -23,7 +25,7 @@ import paddle
from paddle import fluid from paddle import fluid
from paddle.fluid import core from paddle.fluid import core
'''
@param.parameterized_class( @param.parameterized_class(
('shape1', 'shape2'), ('shape1', 'shape2'),
[ [
...@@ -118,7 +120,6 @@ class TestAddHighGradCheck(unittest.TestCase): ...@@ -118,7 +120,6 @@ class TestAddHighGradCheck(unittest.TestCase):
for p in places: for p in places:
self.func_double(p) self.func_double(p)
self.func_triple(p) self.func_triple(p)
'''
@param.parameterized_class( @param.parameterized_class(
...@@ -328,6 +329,7 @@ class TestMultiplyHighGradCheck(unittest.TestCase): ...@@ -328,6 +329,7 @@ class TestMultiplyHighGradCheck(unittest.TestCase):
for p in places: for p in places:
self.func_double(p) self.func_double(p)
self.func_triple(p) self.func_triple(p)
''' '''
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册