Makefile-vllm 480 Bytes
Newer Older
1
commit_rocm := 4e0929e6e4fa0a3d09d358715c288020ea9dc247
2

3
4
5
build-vllm-rocm:
	if [ ! -d 'vllm' ]; then \
		pip install -U ninja packaging --no-cache-dir && \
6
		git clone https://github.com/mht-sharma/vllm.git vllm; \
7
	fi
8
9
	cd vllm && git fetch && git checkout $(commit_rocm) &&  \
	PYTORCH_ROCM_ARCH="gfx90a;gfx942" python setup.py build
10

Nicolas Patry's avatar
Nicolas Patry committed
11
install-vllm-rocm: build-vllm-rocm
12
	cd vllm && git fetch && git checkout $(commit_rocm) && \
13
	PYTORCH_ROCM_ARCH="gfx90a;gfx942" pip install -e .