diff --git a/test/test_network_ops/test_flip.py b/test/test_network_ops/test_flip.py index c77f0ee1b63f14c2054d1a667751c585b88fb587..11d2800912e131ee046a55bdbdd3a7dd4c71c4ee 100644 --- a/test/test_network_ops/test_flip.py +++ b/test/test_network_ops/test_flip.py @@ -15,6 +15,7 @@ import torch import torch_npu import numpy as np + from torch_npu.testing.common_utils import TestCase, run_tests from torch_npu.testing.common_device_type import instantiate_device_type_tests from torch_npu.testing.util_test import create_common_tensor diff --git a/test/test_network_ops/prelu.py b/test/test_network_ops/test_prelu.py similarity index 99% rename from test/test_network_ops/prelu.py rename to test/test_network_ops/test_prelu.py index cebcf1f1e85f1357d0a979867176f077b6ad8efd..6a41dc94fafabc03a8e3a51b7a9a8f42e29d0acb 100644 --- a/test/test_network_ops/prelu.py +++ b/test/test_network_ops/test_prelu.py @@ -15,6 +15,7 @@ import torch import torch_npu import numpy as np + from torch_npu.testing.common_utils import TestCase, run_tests from torch_npu.testing.common_device_type import instantiate_device_type_tests from torch_npu.testing.util_test import create_common_tensor diff --git a/test/test_network_ops/prelu_backward.py b/test/test_network_ops/test_prelu_backward.py similarity index 99% rename from test/test_network_ops/prelu_backward.py rename to test/test_network_ops/test_prelu_backward.py index 57188906abffa81272fdde82521c0a7f6cb916ee..5b219038d528b2833831bb60c25061f88073fd81 100644 --- a/test/test_network_ops/prelu_backward.py +++ b/test/test_network_ops/test_prelu_backward.py @@ -15,6 +15,7 @@ import torch import torch_npu import numpy as np + from torch_npu.testing.common_utils import TestCase, run_tests from torch_npu.testing.common_device_type import instantiate_device_type_tests from torch_npu.testing.util_test import create_common_tensor