feat: initial commit
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
PREFIX ?= $(HOME)/.local
|
||||
BINDIR ?= $(PREFIX)/bin
|
||||
PROG ?= video2srt
|
||||
PYTHON ?= python3
|
||||
VENV ?= .venv
|
||||
BACKEND ?= rocm
|
||||
|
||||
PIP := $(VENV)/bin/pip
|
||||
|
||||
ifeq ($(BACKEND),rocm)
|
||||
TORCH_INDEX := --index-url https://download.pytorch.org/whl/rocm6.3
|
||||
else ifeq ($(BACKEND),cuda)
|
||||
TORCH_INDEX := --index-url https://download.pytorch.org/whl/cu124
|
||||
else ifeq ($(BACKEND),cpu)
|
||||
TORCH_INDEX :=
|
||||
else
|
||||
$(error Unknown BACKEND=$(BACKEND), expected: rocm | cuda | cpu)
|
||||
endif
|
||||
|
||||
.PHONY: build install clean uninstall
|
||||
|
||||
build: $(VENV)/.installed
|
||||
|
||||
$(VENV)/.torch-$(BACKEND): | $(VENV)
|
||||
$(PIP) install torch torchaudio $(TORCH_INDEX)
|
||||
touch $@
|
||||
|
||||
$(VENV)/.installed: pyproject.toml $(VENV)/.torch-$(BACKEND)
|
||||
$(PIP) install -e .
|
||||
ifeq ($(BACKEND),rocm)
|
||||
$(PIP) uninstall -y triton || true
|
||||
$(PIP) install --force-reinstall --no-deps pytorch-triton-rocm $(TORCH_INDEX)
|
||||
endif
|
||||
touch $@
|
||||
|
||||
$(VENV):
|
||||
$(PYTHON) -m venv $(VENV)
|
||||
|
||||
$(BINDIR)/$(PROG): $(VENV)/.installed
|
||||
install -D -m 755 $(VENV)/bin/$(PROG) $@
|
||||
|
||||
install: $(BINDIR)/$(PROG)
|
||||
|
||||
clean:
|
||||
rm -rf $(VENV)
|
||||
|
||||
uninstall:
|
||||
rm -f $(BINDIR)/$(PROG)
|
||||
Reference in New Issue
Block a user