Code

ROCm support#252

Code

Open

88Ocelot wants to merge 4 commits intosmallcloudai:mainsmallcloudai/refact:mainfrom 88Ocelot:feature/rocm88Ocelot/refact:feature/rocmCopy head branch name to clipboard

+166-6

Commits

Commits on Dec 25, 2023

• Add docker-compose for ROCm

  88Ocelot

  authored
  bb3133a

  Copy full SHA for bb3133a

• Add Dockerfile for ROCm

  88Ocelot

  authored
  db42080

  Copy full SHA for db42080

• Configure setup.py for ROCm and CUDA support

  88Ocelot

  authored
  65c50d0

  Copy full SHA for 65c50d0

• Add support for rocm-smi

  88Ocelot

  authored
  d5a8e72

  Copy full SHA for d5a8e72