Skip to content

'triton.language.extra.tang.libdevice' has no attribute 'log2' #5347

Description

@tengqm

Error output:

$ python
Python 3.10.20 (main, Jun 23 2026, 15:20:26) [Clang 22.1.3 ] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import flag_gems
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/secure/FlagGems/.venv/lib/python3.10/site-packages/flag_gems/__init__.py", line 25, in <module>
    from flag_gems.fused import *  # noqa: F403
  File "/home/secure/FlagGems/.venv/lib/python3.10/site-packages/flag_gems/runtime/backend/_sunrise/__init__.py", line 625, in import_with_sunrise_pointwise_patch
    module = original_import(name, globals, locals, fromlist, level)
  File "/home/secure/FlagGems/.venv/lib/python3.10/site-packages/flag_gems/fused/__init__.py", line 91, in <module>
    from flag_gems.fused.outer import outer
  File "/home/secure/FlagGems/.venv/lib/python3.10/site-packages/flag_gems/fused/outer.py", line 19, in <module>
    from flag_gems.ops import mul, mv
  File "/home/secure/FlagGems/.venv/lib/python3.10/site-packages/flag_gems/ops/__init__.py", line 503, in <module>
    from flag_gems.ops.pairwise_distance import pairwise_distance
  File "/home/secure/FlagGems/.venv/lib/python3.10/site-packages/flag_gems/ops/pairwise_distance.py", line 13, in <module>
    log2 = tl_extra_shim.log2
AttributeError: module 'triton.language.extra.tang.libdevice' has no attribute 'log2'. Did you mean: 'log'?
>>> exit()

Metadata

Metadata

Assignees

No one assigned

    Labels

    P0Priority 0 - urgentbugSomething isn't workingvendor/Sunrise

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions