DISABLED test_triton_alltoall (__main__.NVSHMEMTritonTest) #158840
Description
Platforms: linux
This test was disabled because it is failing in CI. See recent examples and the most recent trunk workflow logs.
Over the past 6 hours, it has been determined flaky in 8 workflow(s) with 8 failures and 8 successes.
Debugging instructions (after clicking on the recent samples link):
DO NOT ASSUME THINGS ARE OKAY IF THE CI IS GREEN. We now shield flaky tests from developers so CI will thus be green but it will be harder to parse the logs.
To find relevant log snippets:
- Click on the workflow logs linked above
- Click on the Test step of the job so that it is expanded. Otherwise, the grepping will not work.
- Grep for
test_triton_alltoall - There should be several instances run (as flaky tests are rerun in CI) from which you can study the logs.
Sample error message
Traceback (most recent call last):
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_distributed.py", line 1777, in wrapper
raise rv
RuntimeError: Exception in worker process:
Traceback (most recent call last):
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_distributed.py", line 1642, in _worker_loop
cls._run_test_given_id(test_id)
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_distributed.py", line 1615, in _run_test_given_id
test_fn(**kwargs)
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_distributed.py", line 1786, in wrapper
fn()
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 3223, in wrapper
method(*args, **kwargs)
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 1916, in wrapper
return fn(*args, **kwargs)
File "/var/lib/jenkins/workspace/test/distributed/test_nvshmem_triton.py", line 860, in test_triton_alltoall
nvshmem_lib = nvshmem.enable_triton()
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/distributed/_symmetric_memory/_nvshmem_triton.py", line 33, in enable_triton
raise RuntimeError("NVSHMEM device library not found")
RuntimeError: NVSHMEM device library not found
To execute this test, run the following from the base repo dir:
python test/distributed/test_nvshmem_triton.py NVSHMEMTritonTest.test_triton_alltoall
This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0
Test file path: distributed/test_nvshmem_triton.py
For all disabled tests (by GitHub issue), see https://hud.pytorch.org/disabled.
cc @H-Huang @awgu @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @pragupta @clee2000