llama2_cu_python/Makefile

NVCC = nvcc

.PHONY: libllama2
libllama2: llama2.cu
	$(NVCC) -DUSE_CUDA --shared -O3 -lcublas -lm -o libllama2.so llama2.cu --compiler-options '-fPIC'

.PHONY: clean
clean:
	rm -f libllama2.so