diff --git a/test/test_c10d.py b/test/test_c10d.py index 428280e9430430e36ab24b471c132d6f4dc9fcd2..1046bab6ad14aebe7ac1b29362fc54741f49a84e 100644 --- a/test/test_c10d.py +++ b/test/test_c10d.py @@ -17,10 +17,10 @@ from enum import IntEnum, unique import os import unittest import torch -import torch_npu import torch.distributed as c10d import torch.distributed as dist import torch.multiprocessing as mp +import torch_npu from torch_npu.testing.testcase import TestCase, run_tests