提交 325f0722 编写于 作者: J juncaipeng 提交者: Tao Luo

skip all mkldnn op test (#22363)

上级 aaa4fe49
...@@ -198,7 +198,7 @@ class OpTest(unittest.TestCase): ...@@ -198,7 +198,7 @@ class OpTest(unittest.TestCase):
all_op_kernels = core._get_all_register_op_kernels() all_op_kernels = core._get_all_register_op_kernels()
grad_op = op_type + '_grad' grad_op = op_type + '_grad'
if grad_op in all_op_kernels.keys(): if grad_op in all_op_kernels.keys():
if hasattr(cls, "use_mkldnn") and cls.use_mkldnn == True: if is_mkldnn_op_test():
grad_op_kernels = all_op_kernels[grad_op] grad_op_kernels = all_op_kernels[grad_op]
for grad_op_kernel in grad_op_kernels: for grad_op_kernel in grad_op_kernels:
if 'MKLDNN' in grad_op_kernel: if 'MKLDNN' in grad_op_kernel:
...@@ -207,6 +207,14 @@ class OpTest(unittest.TestCase): ...@@ -207,6 +207,14 @@ class OpTest(unittest.TestCase):
return False return False
return True return True
def is_mkldnn_op_test():
if (hasattr(cls, "use_mkldnn") and cls.use_mkldnn == True) or \
(hasattr(cls, "attrs") and "use_mkldnn" in cls.attrs and \
cls.attrs["use_mkldnn"] == True):
return True
else:
return False
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, "
...@@ -226,7 +234,7 @@ class OpTest(unittest.TestCase): ...@@ -226,7 +234,7 @@ class OpTest(unittest.TestCase):
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): and not is_mkldnn_op_test():
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.
先完成此消息的编辑!
想要评论请 注册