Skip to content

[Fix] Drop deprecated tl.extra.cuda.libdevice prefix - #1005

Merged
zhiyuan1i merged 1 commit into
fla-org:mainfrom
kashif:fix-tl-extra-cuda-deprecation
Jul 6, 2026
Merged

zhiyuan1i merged 1 commit into
fla-org:mainfrom
kashif:fix-tl-extra-cuda-deprecation

Conversation

@kashif

@kashif kashif commented Jul 5, 2026

Copy link
Copy Markdown
Contributor

mainline triton dropped implicit conversion for `tl.extra.cuda.` (the old CUDA-namespaced spelling). recent nightly builds now raise:

```
RuntimeError: Implicit conversion of CUDA __nv_roundf device function has been dropped;
please, update your source program to use triton.language.extra.
to replace triton.language.extra.cuda.
```

which currently breaks bitnet at import/compile time. two call sites in `fla/modules/fused_bitlinear.py`, both switched to `tl.extra.libdevice.round` (backend-agnostic).

verified on rocm7.2 + torch 2.14 nightly with a source-built triton at the latest main.

mainline triton dropped the implicit CUDA->extra conversion, so
tl.extra.cuda.libdevice.round now raises. use tl.extra.libdevice.round
which is backend-agnostic.
@zhiyuan1i
zhiyuan1i merged commit 29f1a93 into fla-org:main Jul 6, 2026
9 of 10 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants