pytorch - 💡(How to fix) Fix [Triton][Triton-Ext] Include TLX Triton Extension [2 comments, 2 participants]

New Feature for Release

We would like to include the Triton Extensions (https://github.com/triton-lang/triton-ext) into the Triton PyTorch release. This would mainly require building Triton with TRITON_EXT_ENABLED=1; and then including the triton-ext repo as it's own package. For the moment we are really only concern with including the TLX extensions https://github.com/triton-lang/triton-ext/tree/main/extensions/utlx so we do not need to include the entire triton-ext repo.

Point(s) of contact

crobeck, plotfi

Release Mode (pytorch/pytorch features only)

Out-of-tree

Out-Of-Tree Repo

https://github.com/triton-lang/triton-ext/

Description and value to the user

The main value add, specifically to Meta is to ship TLX as a fully out of tree extension to upstream Triton (https://github.com/triton-lang/triton-ext/tree/main/extensions/utlx)

Link to design doc, GitHub issues, past submissions, etc

No response

What feedback adopters have provided

No response

Plan for documentations / tutorials

Tutorial exists

Additional context for tutorials

No response

Marketing/Blog Coverage

Yes

Are you requesting other marketing assistance with this feature?

No response

Release Version

No response

OS / Platform / Compute Coverage

No response

Testing Support (CI, test cases, etc..)

No response

cc @seemethere @malfet @atalman @tinglvv @nWEIdia @bertmaher @int3 @davidberard98 @nmacchioni @embg @peterbell10 @aakhundov @jataylo @iupaikov-amd