[AMD] Add mori-shmem backend support by jhchouuu · Pull Request #145 · ByteDance-Seed/Triton-distributed (original) (raw)
added 7 commits
- Add libmori_shmem_device Python bindings for Triton kernels
- Add comprehensive test suite for mori-shmem APIs (PyTorch/UniqueID init)
- Add build script (build_mori_shmem.sh) for mori and device BC
Copilot AI review requested due to automatic review settings
Copilot AI review requested due to automatic review settings
Copilot AI review requested due to automatic review settings
Feature: add more mori_shmem bitcode wrappers && add mori_shmem bandwith test
Feature: mori_shmem supoort dl op call && small refactor
fix ci && move tensor create to mori library
Refine setup.py
Copilot AI review requested due to automatic review settings
Copilot AI review requested due to automatic review settings
Copilot AI review requested due to automatic review settings
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.Learn more about bidirectional Unicode characters
[ Show hidden characters]({{ revealButtonHref }})