From 7a070920a04a3ce843fe2b87d2790341b6af7080 Mon Sep 17 00:00:00 2001 From: yuhaiyan Date: Mon, 7 Jul 2025 12:01:02 +0000 Subject: [PATCH 1/3] update test/npu/_fault_mode_cases/error_set_device.py. Signed-off-by: yuhaiyan --- test/npu/_fault_mode_cases/error_set_device.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/npu/_fault_mode_cases/error_set_device.py b/test/npu/_fault_mode_cases/error_set_device.py index ddda33f346..d25a0f6525 100644 --- a/test/npu/_fault_mode_cases/error_set_device.py +++ b/test/npu/_fault_mode_cases/error_set_device.py @@ -11,7 +11,7 @@ def _worker(i: int) -> None: def set_device(): torch_npu.npu.set_device(0) multiprocessing.set_start_method("spawn", force=True) - jobs = [multiprocessing.Process(target=_worker, args=(i,)) for i in range(70)] + jobs = [multiprocessing.Process(target=_worker, args=(i,)) for i in range(30)] for p in jobs: p.start() -- Gitee From 5213d74eb44788ef75be81f449516a9d8a99e24d Mon Sep 17 00:00:00 2001 From: yuhaiyan Date: Tue, 8 Jul 2025 01:24:23 +0000 Subject: [PATCH 2/3] update test/npu/test_fault_mode.py. Signed-off-by: yuhaiyan --- test/npu/test_fault_mode.py | 1 - 1 file changed, 1 deletion(-) diff --git a/test/npu/test_fault_mode.py b/test/npu/test_fault_mode.py index 88bc8cca19..df7650eb4e 100644 --- a/test/npu/test_fault_mode.py +++ b/test/npu/test_fault_mode.py @@ -1,6 +1,5 @@ import os import subprocess - import torch from torch.testing._internal.common_utils import TestCase, run_tests from torch.utils.checkpoint import checkpoint -- Gitee From df8df2a2ed2ed881bf94dc6f686c74b7dc041a36 Mon Sep 17 00:00:00 2001 From: yuhaiyan Date: Tue, 8 Jul 2025 07:50:11 +0000 Subject: [PATCH 3/3] update test/npu/_fault_mode_cases/error_set_device.py. Signed-off-by: yuhaiyan --- test/npu/_fault_mode_cases/error_set_device.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/npu/_fault_mode_cases/error_set_device.py b/test/npu/_fault_mode_cases/error_set_device.py index d25a0f6525..1545ee9123 100644 --- a/test/npu/_fault_mode_cases/error_set_device.py +++ b/test/npu/_fault_mode_cases/error_set_device.py @@ -11,7 +11,7 @@ def _worker(i: int) -> None: def set_device(): torch_npu.npu.set_device(0) multiprocessing.set_start_method("spawn", force=True) - jobs = [multiprocessing.Process(target=_worker, args=(i,)) for i in range(30)] + jobs = [multiprocessing.Process(target=_worker, args=(i,)) for i in range(100)] for p in jobs: p.start() @@ -20,4 +20,5 @@ def set_device(): p.join() -set_device() +if __name__ == "__main__": + set_device() -- Gitee