Skip to content

Commit 4c884ca

Browse files
committed
Add log for UT random issues to debug
Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
1 parent 04e5a38 commit 4c884ca

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

test/xpu/test_ops_xpu.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,12 @@
11
# Owner(s): ["module: intel"]
22

33

4-
from torch.testing._internal.common_device_type import instantiate_device_type_tests
5-
from torch.testing._internal.common_utils import run_tests
4+
import torch
5+
import torch._prims as prims
6+
import torch.utils._pytree as pytree
7+
import warnings
8+
from torch.testing._internal.common_device_type import instantiate_device_type_tests, onlyXPU, OpDTypes, ops
9+
from torch.testing._internal.common_utils import run_tests, slowTest, suppress_warnings
610

711
try:
812
from xpu_test_utils import XPUPatchForImport

0 commit comments

Comments
 (0)