Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Triton test failed in CI: ImportError: cannot import name 'AttrsDescriptor' #8820

Open
tengyifei opened this issue Mar 12, 2025 · 1 comment · May be fixed by #8825
Open

Triton test failed in CI: ImportError: cannot import name 'AttrsDescriptor' #8820

tengyifei opened this issue Mar 12, 2025 · 1 comment · May be fixed by #8825
Assignees
Labels
CI CI related change testing Testing and coverage related issues.

Comments

@tengyifei
Copy link
Collaborator

See https://github.com/pytorch/xla/actions/runs/13799623710/job/38601675184?pr=8807

======================================================================
ERROR: test_gpu_custom_call_triton_add (__main__.TritonTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/__w/xla/xla/pytorch/xla/test/test_triton.py", line 256, in test_gpu_custom_call_triton_add
    payload = xla_triton.triton_call(
  File "/usr/local/lib/python3.10/site-packages/torch_xla/experimental/triton.py", line 211, in triton_call
    return triton_kernel_call_lowering(array_args, kernel, compiled_kernel,
  File "/usr/local/lib/python3.10/site-packages/torch_xla/experimental/triton.py", line 142, in triton_kernel_call_lowering
    from triton.backends.compiler import AttrsDescriptor
ImportError: cannot import name 'AttrsDescriptor' from 'triton.backends.compiler' (/usr/local/lib/python3.10/site-packages/triton/backends/compiler.py)

======================================================================
ERROR: test_gpu_custom_call_triton_flash_attention (__main__.TritonTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/__w/xla/xla/pytorch/xla/test/test_triton.py", line 292, in test_gpu_custom_call_triton_flash_attention
    payload = xla_triton.triton_call(
  File "/usr/local/lib/python3.10/site-packages/torch_xla/experimental/triton.py", line 211, in triton_call
    return triton_kernel_call_lowering(array_args, kernel, compiled_kernel,
  File "/usr/local/lib/python3.10/site-packages/torch_xla/experimental/triton.py", line 142, in triton_kernel_call_lowering
    from triton.backends.compiler import AttrsDescriptor
ImportError: cannot import name 'AttrsDescriptor' from 'triton.backends.compiler' (/usr/local/lib/python3.10/site-packages/triton/backends/compiler.py)

@ysiraichi ysiraichi added testing Testing and coverage related issues. CI CI related change labels Mar 12, 2025
@ysiraichi ysiraichi linked a pull request Mar 12, 2025 that will close this issue
@ysiraichi
Copy link
Collaborator

Apparently, this was caused by a triton pin update inside PyTorch pytorch/pytorch#148971. This wouldn't have happened if we did have a PyTorch pin, though. Somewhat related to #8745.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CI CI related change testing Testing and coverage related issues.
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants