diff --git a/python/paddle/fluid/tests/unittests/xpu/get_test_cover_info.py b/python/paddle/fluid/tests/unittests/xpu/get_test_cover_info.py index 0c3056ca8a8b4d70beda991c10d3a964d0ac131c..d92378f60f5789cec5776660f45d92990e8a16e0 100644 --- a/python/paddle/fluid/tests/unittests/xpu/get_test_cover_info.py +++ b/python/paddle/fluid/tests/unittests/xpu/get_test_cover_info.py @@ -83,7 +83,7 @@ type_dict_str_to_numpy = { } xpu_test_op_white_list = [] -xpu_test_type_white_list = ['float64'] +xpu_test_device_type_white_list = ['xpu1_float64'] xpu_test_op_type_white_list = ['dropout_float16', 'dropout_grad_float16'] xpu_test_device_op_white_list = [] xpu_test_device_op_type_white_list = [] @@ -106,7 +106,18 @@ def get_op_white_list(): def get_type_white_list(): - type_white_list = xpu_test_type_white_list + xpu_version = core.get_xpu_device_version(0) + version_str = "xpu2" if xpu_version == core.XPUVersion.XPU2 else "xpu1" + xpu1_type_white_list = [] + xpu2_type_white_list = [] + for device_type in xpu_test_device_type_white_list: + device, t_type = device_type.split("_") + if "xpu1" == device: + xpu1_type_white_list.append(t_type) + else: + xpu2_type_white_list.append(t_type) + + type_white_list = xpu1_type_white_list if version_str == "xpu1" else xpu2_type_white_list if os.getenv('XPU_TEST_TYPE_WHITE_LIST') is not None: type_white_list.extend( os.getenv('XPU_TEST_TYPE_WHITE_LIST').strip().split(','))