This repository was archived by the owner on Jul 8, 2026. It is now read-only.

Support AMD MI300 GPU - #155

Open

Yuxiang Yang (tocean) wants to merge 14 commits into

mainfrom

Commits on Mar 6, 2024

dist_op support amd
wenchenvincent
committed
e23f9d0
clean code

Yuxiang Yang (tocean)
authored and
wenchenvincent
committed
8dc36e5
Move cpp files to csrc folder

Yuxiang Yang (tocean)
authored and
wenchenvincent
committed
90f70fe
rename dist to dist_op

Yuxiang Yang (tocean)
authored and
wenchenvincent
committed
94af5c6
remove unless include

Yuxiang Yang (tocean)
authored and
wenchenvincent
committed
cfd7286
make extension compilable with AMD GPU

Yuxiang Yang (tocean)
authored and
wenchenvincent
committed
b7d201b
build succeed on Nvidia GPU

Yuxiang Yang (tocean)
authored and
wenchenvincent
committed
0fb635a
add optional dependencies in pyproject.toml

Yuxiang Yang (tocean)
authored and
wenchenvincent
committed
cd9b482
UT
wenchenvincent
committed
33f726a

Commits on Mar 7, 2024

make all non-distributed UT pass
wenchenvincent
committed
363cb01

Commits on Mar 11, 2024

fix bug in dist op
wenchenvincent
committed
1efec4a
fix ut bugs
wenchenvincent
committed
f815059

Commits on Mar 13, 2024

add rccl as sumodule
Yuxiang Yang (tocean)
committed
d3d80cf
remove rccl
Yuxiang Yang (tocean)
committed
d4404c7