Does colossalai support rocm?

hpcaitech / ColossalAI

Making large AI models cheaper, faster and more accessible

https://www.colossalai.org

Apache License 2.0

38.67k stars 4.33k forks source link

Open zhaohm14 opened 5 months ago

zhaohm14 commented 5 months ago

Does colossalai support rocm? Thanks!

Edenzzzz commented 4 months ago

Not currently, but some collaboration with AMD is underway

unclemusclez commented 3 months ago

any update?