From ebb4848a2dd48928f4212b702ef5d6168410bd53 Mon Sep 17 00:00:00 2001 From: Xiongfei Wei Date: Thu, 25 Jan 2024 20:27:26 +0000 Subject: [PATCH 01/11] re-apply https://github.com/pytorch/xla/pull/6247 --- torch_xla/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index 03e49a60415..c5807a97145 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -147,6 +147,8 @@ def _setup_tpu_vm_library_path() -> bool: def _prepare_to_exit(): + device = _XLAC._xla_get_default_device() + _XLAC._set_all_reduce_token(device, None) _XLAC._prepare_to_exit() if int(os.environ.get('PT_XLA_DEBUG', '0')): _summarize_fn_tracker() From 01735f32aa89c7d703e4b52b9b31a161d9678480 Mon Sep 17 00:00:00 2001 From: Xiongfei Wei Date: Fri, 26 Jan 2024 00:56:06 +0000 Subject: [PATCH 02/11] make the test shorter --- .github/workflows/_test.yml | 38 ++++++++++++++++++------------------- test/run_tests.sh | 28 --------------------------- 2 files changed, 19 insertions(+), 47 deletions(-) diff --git a/.github/workflows/_test.yml b/.github/workflows/_test.yml index cd5e8f9bb94..323cdbfbca6 100644 --- a/.github/workflows/_test.yml +++ b/.github/workflows/_test.yml @@ -50,34 +50,34 @@ jobs: matrix: include: # Use readable strings as they define the workflow titles. - - run_benchmark_tests: 'benchmark_tests' - - run_cpp_tests1: 'cpp_tests1' - - run_cpp_tests2: 'cpp_tests2' + # - run_benchmark_tests: 'benchmark_tests' + # - run_cpp_tests1: 'cpp_tests1' + # - run_cpp_tests2: 'cpp_tests2' - run_python_tests: 'python_tests' run_xla_op_tests1: 'xla_op1' - - run_python_tests: 'python_tests' - run_xla_op_tests2: 'xla_op2' - - run_python_tests: 'python_tests' - run_xla_op_tests3: 'xla_op3' - - run_python_tests: 'python_tests' - run_torch_mp_op_tests: 'torch_mp_op' + # - run_python_tests: 'python_tests' + # run_xla_op_tests2: 'xla_op2' + # - run_python_tests: 'python_tests' + # run_xla_op_tests3: 'xla_op3' + # - run_python_tests: 'python' + # run_torch_mp_op_tests: 'torch_mp_op' timeout-minutes: ${{ inputs.timeout-minutes }} env: DOCKER_IMAGE: ${{ inputs.docker-image }} WORKDIR: /var/lib/jenkins/workspace GCLOUD_SERVICE_KEY: ${{ secrets.gcloud-service-key }} USE_COVERAGE: ${{ inputs.collect-coverage && '1' || '0' }} - XLA_SKIP_XRT_TESTS: ${{ inputs.disable-xrt }} - XLA_SKIP_TORCH_OP_TESTS: ${{ inputs.disable-pjrt }} - XLA_SKIP_MP_OP_TESTS: ${{ inputs.disable-pjrt }} - RUN_BENCHMARK_TESTS: ${{ matrix.run_benchmark_tests }} - RUN_CPP_TESTS1: ${{ matrix.run_cpp_tests1 }} - RUN_CPP_TESTS2: ${{ matrix.run_cpp_tests2 }} + # XLA_SKIP_XRT_TESTS: ${{ inputs.disable-xrt }} + # XLA_SKIP_TORCH_OP_TESTS: ${{ inputs.disable-pjrt }} + # XLA_SKIP_MP_OP_TESTS: ${{ inputs.disable-pjrt }} + # RUN_BENCHMARK_TESTS: ${{ matrix.run_benchmark_tests }} + # RUN_CPP_TESTS1: ${{ matrix.run_cpp_tests1 }} + # RUN_CPP_TESTS2: ${{ matrix.run_cpp_tests2 }} RUN_PYTHON_TESTS: ${{ matrix.run_python_tests }} RUN_XLA_OP_TESTS1: ${{ matrix.run_xla_op_tests1 }} - RUN_XLA_OP_TESTS2: ${{ matrix.run_xla_op_tests2 }} - RUN_XLA_OP_TESTS3: ${{ matrix.run_xla_op_tests3 }} - RUN_TORCH_MP_OP_TESTS: ${{ matrix.run_torch_mp_op_tests }} + # RUN_XLA_OP_TESTS2: ${{ matrix.run_xla_op_tests2 }} + # RUN_XLA_OP_TESTS3: ${{ matrix.run_xla_op_tests3 }} + # RUN_TORCH_MP_OP_TESTS: ${{ matrix.run_torch_mp_op_tests }} steps: - name: Setup Linux uses: pytorch/test-infra/.github/actions/setup-linux@main @@ -112,7 +112,7 @@ jobs: run: | echo "DOCKER_IMAGE: ${DOCKER_IMAGE}" docker pull "${DOCKER_IMAGE}" - pid=$(docker run --shm-size=16g ${GPU_FLAG:-} -e USE_COVERAGE -e XLA_SKIP_XRT_TESTS -e XLA_SKIP_TORCH_OP_TESTS -e XLA_SKIP_MP_OP_TESTS -e RUN_BENCHMARK_TESTS -e RUN_CPP_TESTS1 -e RUN_CPP_TESTS2 -e RUN_PYTHON_TESTS -e RUN_XLA_OP_TESTS1 -e RUN_XLA_OP_TESTS2 -e RUN_XLA_OP_TESTS3 -e RUN_TORCH_MP_OP_TESTS -t -d -w "$WORKDIR" "${DOCKER_IMAGE}") + pid=$(docker run --shm-size=16g ${GPU_FLAG:-} -e USE_COVERAGE -e RUN_PYTHON_TESTS -e RUN_XLA_OP_TESTS1 -t -d -w "$WORKDIR" "${DOCKER_IMAGE}") echo "${GCLOUD_SERVICE_KEY}" | docker exec -i "${pid}" sh -c "cat >> /tmp/pytorch/xla/default_credentials.json" echo "pid=${pid}" >> "${GITHUB_ENV}" - name: Test diff --git a/test/run_tests.sh b/test/run_tests.sh index 84debb84b25..75bc606b620 100755 --- a/test/run_tests.sh +++ b/test/run_tests.sh @@ -154,38 +154,10 @@ function run_torch_op_tests { # DO NOT MODIFY function run_xla_op_tests1 { - run_dynamic "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY - run_dynamic "$CDIR/ds/test_dynamic_shapes.py" - run_dynamic "$CDIR/ds/test_dynamic_shape_models.py" "$@" --verbosity=$VERBOSITY - run_eager_debug "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY - run_test "$CDIR/test_grad_checkpoint.py" - run_test "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY - run_test_without_functionalization "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY - run_pt_xla_debug "$CDIR/debug_tool/test_pt_xla_debug.py" - run_test "$CDIR/test_async_closures.py" - run_test "$CDIR/test_hlo_metadata.py" - run_test "$CDIR/test_profiler.py" run_test "$CDIR/pjrt/test_runtime.py" run_test "$CDIR/pjrt/test_runtime_single_proc_gpu.py" run_test "$CDIR/pjrt/test_runtime_multi_gpu.py" run_test "$CDIR/pjrt/test_runtime_multi_cpu.py" - run_test "$CDIR/pjrt/test_internal_tpu.py" - run_test "$CDIR/pjrt/test_ddp.py" - run_test "$CDIR/pjrt/test_mesh_service.py" - run_test "$CDIR/test_ops.py" - run_test "$CDIR/test_metrics.py" - run_test "$CDIR/test_zero1.py" - run_test "$CDIR/dynamo/test_dynamo_integrations_util.py" - run_test "$CDIR/dynamo/test_dynamo.py" - run_test "$CDIR/dynamo/test_bridge.py" - run_test "$CDIR/dynamo/test_num_output.py" - run_save_tensor_ir "$CDIR/dynamo/test_dynamo_graph_dump.py" - run_use_bf16 "$CDIR/test_data_type.py" - run_xla_ir_debug "$CDIR/test_env_var_mapper.py" - run_xla_hlo_debug "$CDIR/test_env_var_mapper.py" - run_xla_hlo_debug "$CDIR/stablehlo/test_stablehlo_save_load.py" - run_save_tensor_ir "$CDIR/spmd/test_spmd_graph_dump.py" - run_save_tensor_hlo "$CDIR/spmd/test_spmd_graph_dump.py" } # DO NOT MODIFY From fdc8e9b920ebef8f271119ac03e93a014f4c44c7 Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Mon, 5 Feb 2024 21:52:02 +0000 Subject: [PATCH 03/11] set the exit code to 1 and let's check the ci status --- torch_xla/__init__.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index c5807a97145..1112b299aec 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -147,11 +147,15 @@ def _setup_tpu_vm_library_path() -> bool: def _prepare_to_exit(): - device = _XLAC._xla_get_default_device() - _XLAC._set_all_reduce_token(device, None) - _XLAC._prepare_to_exit() - if int(os.environ.get('PT_XLA_DEBUG', '0')): - _summarize_fn_tracker() + try: + device = _XLAC._xla_get_default_device() + _XLAC._set_all_reduce_token(device, None) + _XLAC._prepare_to_exit() + if int(os.environ.get('PT_XLA_DEBUG', '0')): + _summarize_fn_tracker() + except Exception as e: + print('Caught an exception in the atexit callback: ', e) + os._exit(1) def _init_xla_lazy_backend(): From a2c4167dd1a1c84e79efd3c9c42acf3c997fe4c0 Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Mon, 5 Feb 2024 23:23:01 +0000 Subject: [PATCH 04/11] make sure we flush printing the exception --- torch_xla/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index 1112b299aec..be700ae917f 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -154,7 +154,7 @@ def _prepare_to_exit(): if int(os.environ.get('PT_XLA_DEBUG', '0')): _summarize_fn_tracker() except Exception as e: - print('Caught an exception in the atexit callback: ', e) + print('Caught an exception in the atexit callback: ', e, flush=True) os._exit(1) From cba8e5d6d70d275945e9f551939de5a722411dab Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Tue, 6 Feb 2024 21:40:27 +0000 Subject: [PATCH 05/11] revert the repro change --- .github/workflows/_test.yml | 38 ++++++++++++++++++------------------- test/run_tests.sh | 28 +++++++++++++++++++++++++++ torch_xla/__init__.py | 2 -- 3 files changed, 47 insertions(+), 21 deletions(-) diff --git a/.github/workflows/_test.yml b/.github/workflows/_test.yml index 323cdbfbca6..49ebe09a6e4 100644 --- a/.github/workflows/_test.yml +++ b/.github/workflows/_test.yml @@ -50,34 +50,34 @@ jobs: matrix: include: # Use readable strings as they define the workflow titles. - # - run_benchmark_tests: 'benchmark_tests' - # - run_cpp_tests1: 'cpp_tests1' - # - run_cpp_tests2: 'cpp_tests2' + - run_benchmark_tests: 'benchmark_tests' + - run_cpp_tests1: 'cpp_tests1' + - run_cpp_tests2: 'cpp_tests2' - run_python_tests: 'python_tests' run_xla_op_tests1: 'xla_op1' - # - run_python_tests: 'python_tests' - # run_xla_op_tests2: 'xla_op2' - # - run_python_tests: 'python_tests' - # run_xla_op_tests3: 'xla_op3' - # - run_python_tests: 'python' - # run_torch_mp_op_tests: 'torch_mp_op' + - run_python_tests: 'python_tests' + run_xla_op_tests2: 'xla_op2' + - run_python_tests: 'python_tests' + run_xla_op_tests3: 'xla_op3' + - run_python_tests: 'python' + run_torch_mp_op_tests: 'torch_mp_op' timeout-minutes: ${{ inputs.timeout-minutes }} env: DOCKER_IMAGE: ${{ inputs.docker-image }} WORKDIR: /var/lib/jenkins/workspace GCLOUD_SERVICE_KEY: ${{ secrets.gcloud-service-key }} USE_COVERAGE: ${{ inputs.collect-coverage && '1' || '0' }} - # XLA_SKIP_XRT_TESTS: ${{ inputs.disable-xrt }} - # XLA_SKIP_TORCH_OP_TESTS: ${{ inputs.disable-pjrt }} - # XLA_SKIP_MP_OP_TESTS: ${{ inputs.disable-pjrt }} - # RUN_BENCHMARK_TESTS: ${{ matrix.run_benchmark_tests }} - # RUN_CPP_TESTS1: ${{ matrix.run_cpp_tests1 }} - # RUN_CPP_TESTS2: ${{ matrix.run_cpp_tests2 }} + XLA_SKIP_XRT_TESTS: ${{ inputs.disable-xrt }} + XLA_SKIP_TORCH_OP_TESTS: ${{ inputs.disable-pjrt }} + XLA_SKIP_MP_OP_TESTS: ${{ inputs.disable-pjrt }} + RUN_BENCHMARK_TESTS: ${{ matrix.run_benchmark_tests }} + RUN_CPP_TESTS1: ${{ matrix.run_cpp_tests1 }} + RUN_CPP_TESTS2: ${{ matrix.run_cpp_tests2 }} RUN_PYTHON_TESTS: ${{ matrix.run_python_tests }} RUN_XLA_OP_TESTS1: ${{ matrix.run_xla_op_tests1 }} - # RUN_XLA_OP_TESTS2: ${{ matrix.run_xla_op_tests2 }} - # RUN_XLA_OP_TESTS3: ${{ matrix.run_xla_op_tests3 }} - # RUN_TORCH_MP_OP_TESTS: ${{ matrix.run_torch_mp_op_tests }} + RUN_XLA_OP_TESTS2: ${{ matrix.run_xla_op_tests2 }} + RUN_XLA_OP_TESTS3: ${{ matrix.run_xla_op_tests3 }} + RUN_TORCH_MP_OP_TESTS: ${{ matrix.run_torch_mp_op_tests }} steps: - name: Setup Linux uses: pytorch/test-infra/.github/actions/setup-linux@main @@ -112,7 +112,7 @@ jobs: run: | echo "DOCKER_IMAGE: ${DOCKER_IMAGE}" docker pull "${DOCKER_IMAGE}" - pid=$(docker run --shm-size=16g ${GPU_FLAG:-} -e USE_COVERAGE -e RUN_PYTHON_TESTS -e RUN_XLA_OP_TESTS1 -t -d -w "$WORKDIR" "${DOCKER_IMAGE}") + pid=$(docker run --shm-size=16g ${GPU_FLAG:-} -e USE_COVERAGE -e XLA_SKIP_XRT_TESTS -e XLA_SKIP_TORCH_OP_TESTS -e XLA_SKIP_MP_OP_TESTS -e RUN_BENCHMARK_TESTS -e RUN_CPP_TESTS1 -e RUN_CPP_TESTS2 -e RUN_PYTHON_TESTS -e RUN_XLA_OP_TESTS1 -e RUN_XLA_OP_TESTS2 -e RUN_XLA_OP_TESTS3 -e RUN_TORCH_MP_OP_TESTS -t -d -w "$WORKDIR" "${DOCKER_IMAGE}") echo "${GCLOUD_SERVICE_KEY}" | docker exec -i "${pid}" sh -c "cat >> /tmp/pytorch/xla/default_credentials.json" echo "pid=${pid}" >> "${GITHUB_ENV}" - name: Test diff --git a/test/run_tests.sh b/test/run_tests.sh index 75bc606b620..84debb84b25 100755 --- a/test/run_tests.sh +++ b/test/run_tests.sh @@ -154,10 +154,38 @@ function run_torch_op_tests { # DO NOT MODIFY function run_xla_op_tests1 { + run_dynamic "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY + run_dynamic "$CDIR/ds/test_dynamic_shapes.py" + run_dynamic "$CDIR/ds/test_dynamic_shape_models.py" "$@" --verbosity=$VERBOSITY + run_eager_debug "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY + run_test "$CDIR/test_grad_checkpoint.py" + run_test "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY + run_test_without_functionalization "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY + run_pt_xla_debug "$CDIR/debug_tool/test_pt_xla_debug.py" + run_test "$CDIR/test_async_closures.py" + run_test "$CDIR/test_hlo_metadata.py" + run_test "$CDIR/test_profiler.py" run_test "$CDIR/pjrt/test_runtime.py" run_test "$CDIR/pjrt/test_runtime_single_proc_gpu.py" run_test "$CDIR/pjrt/test_runtime_multi_gpu.py" run_test "$CDIR/pjrt/test_runtime_multi_cpu.py" + run_test "$CDIR/pjrt/test_internal_tpu.py" + run_test "$CDIR/pjrt/test_ddp.py" + run_test "$CDIR/pjrt/test_mesh_service.py" + run_test "$CDIR/test_ops.py" + run_test "$CDIR/test_metrics.py" + run_test "$CDIR/test_zero1.py" + run_test "$CDIR/dynamo/test_dynamo_integrations_util.py" + run_test "$CDIR/dynamo/test_dynamo.py" + run_test "$CDIR/dynamo/test_bridge.py" + run_test "$CDIR/dynamo/test_num_output.py" + run_save_tensor_ir "$CDIR/dynamo/test_dynamo_graph_dump.py" + run_use_bf16 "$CDIR/test_data_type.py" + run_xla_ir_debug "$CDIR/test_env_var_mapper.py" + run_xla_hlo_debug "$CDIR/test_env_var_mapper.py" + run_xla_hlo_debug "$CDIR/stablehlo/test_stablehlo_save_load.py" + run_save_tensor_ir "$CDIR/spmd/test_spmd_graph_dump.py" + run_save_tensor_hlo "$CDIR/spmd/test_spmd_graph_dump.py" } # DO NOT MODIFY diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index be700ae917f..76d0926df0b 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -148,8 +148,6 @@ def _setup_tpu_vm_library_path() -> bool: def _prepare_to_exit(): try: - device = _XLAC._xla_get_default_device() - _XLAC._set_all_reduce_token(device, None) _XLAC._prepare_to_exit() if int(os.environ.get('PT_XLA_DEBUG', '0')): _summarize_fn_tracker() From cf3742df138490a78d9f9b1913ad2da8b2e304e9 Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Tue, 6 Feb 2024 21:41:31 +0000 Subject: [PATCH 06/11] fix typo --- .github/workflows/_test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/_test.yml b/.github/workflows/_test.yml index 49ebe09a6e4..cd5e8f9bb94 100644 --- a/.github/workflows/_test.yml +++ b/.github/workflows/_test.yml @@ -59,7 +59,7 @@ jobs: run_xla_op_tests2: 'xla_op2' - run_python_tests: 'python_tests' run_xla_op_tests3: 'xla_op3' - - run_python_tests: 'python' + - run_python_tests: 'python_tests' run_torch_mp_op_tests: 'torch_mp_op' timeout-minutes: ${{ inputs.timeout-minutes }} env: From 7471921d314b8cc78265da98cb13de74ed12b206 Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Thu, 8 Feb 2024 18:14:01 +0000 Subject: [PATCH 07/11] fix comment --- torch_xla/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index 76d0926df0b..c7c31e8d12a 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -152,7 +152,7 @@ def _prepare_to_exit(): if int(os.environ.get('PT_XLA_DEBUG', '0')): _summarize_fn_tracker() except Exception as e: - print('Caught an exception in the atexit callback: ', e, flush=True) + logging.error("Caught an exception when exiting the process. Exception: ", exc_info=e) os._exit(1) From 6c9277fcb929fc790795c40f080ce73a6fb576e7 Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Thu, 8 Feb 2024 18:19:49 +0000 Subject: [PATCH 08/11] fix linter --- torch_xla/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index c7c31e8d12a..b848cc34d88 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -152,7 +152,8 @@ def _prepare_to_exit(): if int(os.environ.get('PT_XLA_DEBUG', '0')): _summarize_fn_tracker() except Exception as e: - logging.error("Caught an exception when exiting the process. Exception: ", exc_info=e) + logging.error( + "Caught an exception when exiting the process. Exception: ", exc_info=e) os._exit(1) From e89769d363069f6885806303da8b9489bd72505a Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Thu, 8 Feb 2024 21:42:41 +0000 Subject: [PATCH 09/11] add some comment --- torch_xla/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index b848cc34d88..de0bdea23a0 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -148,13 +148,17 @@ def _setup_tpu_vm_library_path() -> bool: def _prepare_to_exit(): try: + raise ValueError('fake error') _XLAC._prepare_to_exit() if int(os.environ.get('PT_XLA_DEBUG', '0')): _summarize_fn_tracker() except Exception as e: logging.error( "Caught an exception when exiting the process. Exception: ", exc_info=e) - os._exit(1) + # Due to https://bugs.python.org/issue27035, simply raising an exception in the atexit callback does not set the exit code correctly. That is why we need to set the exit code explicitly. + # Using `exit(1)` does not set a correct exit code because it is useful for the interactive interpreter shell and should not be used in programs and it works by raising an exception. (https://docs.python.org/3/library/constants.html#exit) + # sys.exit(1) does not set a correct exit code because it also raises an exception. + exit(1) def _init_xla_lazy_backend(): From bf6686d0ce23f4549a2a0b245073885127cda638 Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Thu, 8 Feb 2024 21:42:54 +0000 Subject: [PATCH 10/11] add some comment --- torch_xla/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index de0bdea23a0..d9e915837fd 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -158,7 +158,7 @@ def _prepare_to_exit(): # Due to https://bugs.python.org/issue27035, simply raising an exception in the atexit callback does not set the exit code correctly. That is why we need to set the exit code explicitly. # Using `exit(1)` does not set a correct exit code because it is useful for the interactive interpreter shell and should not be used in programs and it works by raising an exception. (https://docs.python.org/3/library/constants.html#exit) # sys.exit(1) does not set a correct exit code because it also raises an exception. - exit(1) + os._exit(1) def _init_xla_lazy_backend(): From d22eaf07a1f2ce7c534876c3a8bfbafe11b30bef Mon Sep 17 00:00:00 2001 From: iefgnoix Date: Thu, 8 Feb 2024 23:45:12 +0000 Subject: [PATCH 11/11] fix typo --- torch_xla/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py index d9e915837fd..8e657787ee9 100644 --- a/torch_xla/__init__.py +++ b/torch_xla/__init__.py @@ -148,7 +148,6 @@ def _setup_tpu_vm_library_path() -> bool: def _prepare_to_exit(): try: - raise ValueError('fake error') _XLAC._prepare_to_exit() if int(os.environ.get('PT_XLA_DEBUG', '0')): _summarize_fn_tracker()