未验证 提交 05ee05e2 编写于 作者: J juncaipeng 提交者: GitHub

no fp64 check for mkldnn op test, test=develop (#22217)

上级 f2522e91
...@@ -209,8 +209,8 @@ class OpTest(unittest.TestCase): ...@@ -209,8 +209,8 @@ class OpTest(unittest.TestCase):
if not hasattr(cls, "op_type"): if not hasattr(cls, "op_type"):
raise AssertionError( raise AssertionError(
"This test do not have op_type in class attrs," "This test do not have op_type in class attrs, "
" please set self.__class__.op_type=the_real_op_type manually.") "please set self.__class__.op_type=the_real_op_type manually.")
# case in NO_FP64_CHECK_GRAD_CASES and op in NO_FP64_CHECK_GRAD_OP_LIST should be fixed # case in NO_FP64_CHECK_GRAD_CASES and op in NO_FP64_CHECK_GRAD_OP_LIST should be fixed
if not hasattr(cls, "no_need_check_grad") \ if not hasattr(cls, "no_need_check_grad") \
...@@ -222,9 +222,11 @@ class OpTest(unittest.TestCase): ...@@ -222,9 +222,11 @@ class OpTest(unittest.TestCase):
raise AssertionError("This test of %s op needs check_grad." % raise AssertionError("This test of %s op needs check_grad." %
cls.op_type) cls.op_type)
# check for op test with fp64 precision, but not check mkldnn op test for now
if cls.dtype in [np.float32, np.float64] \ if cls.dtype in [np.float32, np.float64] \
and cls.op_type not in op_accuracy_white_list.NO_FP64_CHECK_GRAD_OP_LIST \ and cls.op_type not in op_accuracy_white_list.NO_FP64_CHECK_GRAD_OP_LIST \
and not hasattr(cls, 'exist_fp64_check_grad'): and not hasattr(cls, 'exist_fp64_check_grad') \
and (not hasattr(cls, "use_mkldnn") or cls.use_mkldnn == False):
raise AssertionError( raise AssertionError(
"This test of %s op needs check_grad with fp64 precision." % "This test of %s op needs check_grad with fp64 precision." %
cls.op_type) cls.op_type)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册