Skip to content

Latest commit

 

History

History
18 lines (14 loc) · 371 Bytes

README.md

File metadata and controls

18 lines (14 loc) · 371 Bytes

TorchAC CUDA/ROCm

For now this is a fork of torchac_cuda Github

This library is needed by LMCache repositories.

Setup

CUDA

python3 setup.py develop

ROCm

HCC_AMDGPU_TARGET=<HIP_ARCHITECTURE> python3 setup.py develop
# e.g. MI210 MI250
HCC_AMDGPU_TARGET=gfx90a python3 setup.py develop