Running the test_run_model test results in a segmentation fault error

[erichuang-cienet]

Environment:

• TPU VM: v6e-8
• Python 3.11
• torch 2.9.0.dev20250901+cpu
• torch-xla 2.9.0+git4c586bd

Command:

export LIBTPU_INIT_ARGS='--xla_tpu_use_enhanced_launch_barrier=false'
pytest -v torchprime/launcher/test_run_model.py

Error:

torchprime/launcher/test_run_model.py::test_run_model_transact Fatal Python error: Segmentation fault

Thread 0x000076568faac800 (most recent call first):
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/torch_xla/core/xla_model.py", line 234 in unlazy
  File "/home/sa_106246016191652103077/torchprime/torchprime/launcher/run_model.py", line 45 in run_model_xla
  File "/home/sa_106246016191652103077/torchprime/torchprime/launcher/test_run_model.py", line 29 in test_run_model_transact
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/python.py", line 159 in pytest_pyfunc_call
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_callers.py", line 121 in _multicall
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_manager.py", line 120 in _hookexec
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_hooks.py", line 512 in __call__
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/python.py", line 1627 in runtest
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/runner.py", line 174 in pytest_runtest_call
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_callers.py", line 121 in _multicall
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_manager.py", line 120 in _hookexec
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_hooks.py", line 512 in __call__
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/runner.py", line 242 in <lambda>
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/runner.py", line 341 in from_call
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/runner.py", line 241 in call_and_report
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/runner.py", line 132 in runtestprotocol
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pytest_forked/__init__.py", line 68 in runforked
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/py/_process/forkedfunc.py", line 65 in _child
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/py/_process/forkedfunc.py", line 50 in __init__
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pytest_forked/__init__.py", line 73 in forked_run_report
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pytest_forked/__init__.py", line 51 in pytest_runtest_protocol
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_callers.py", line 121 in _multicall
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_manager.py", line 120 in _hookexec
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_hooks.py", line 512 in __call__
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/main.py", line 362 in pytest_runtestloop
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_callers.py", line 121 in _multicall
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_manager.py", line 120 in _hookexec
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_hooks.py", line 512 in __call__
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/main.py", line 337 in _main
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/main.py", line 283 in wrap_session
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/main.py", line 330 in pytest_cmdline_main
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_callers.py", line 121 in _multicall
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_manager.py", line 120 in _hookexec
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/pluggy/_hooks.py", line 512 in __call__
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/config/__init__.py", line 175 in main
  File "/home/sa_106246016191652103077/torchprime/venv/lib/python3.11/site-packages/_pytest/config/__init__.py", line 201 in console_main
  File "/home/sa_106246016191652103077/torchprime/venv/bin/pytest", line 7 in <module>

Extension modules: yaml._yaml, numpy._core._multiarray_umath, numpy.linalg._umath_linalg, torch._C, torch._C._dynamo.autograd_compiler, torch._C._dynamo.eval_frame, torch._C._dynamo.guards, torch._C._dynamo.utils, torch._C._fft, torch._C._linalg, torch._C._nested, torch._C._nn, torch._C._sparse, torch._C._special, PIL._imaging, PIL._imagingft, numpy.random._common, numpy.random.bit_generator, numpy.random._bounded_integers, numpy.random._mt19937, numpy.random.mtrand, numpy.random._philox, numpy.random._pcg64, numpy.random._sfc64, numpy.random._generator, jaxlib.cpu_feature_guard, charset_normalizer.md, [requests.packages.charset_normalizer.md](http://requests.packages.charset_normalizer.md/), [requests.packages.chardet.md](http://requests.packages.chardet.md/), markupsafe._speedups (total: 30)
FAILED