search

bnosac / audio.whisper

Transcribe audio files using the "Whisper" Automatic Speech Recognition model from R
Other
113 stars 13 forks source link

Integrate hipBLAS (AMD) #49

Open jwijffels opened 8 months ago

jwijffels commented 8 months ago 
ifdef WHISPER_HIPBLAS
    ROCM_PATH   ?= /opt/rocm
    HIPCC       ?= $(ROCM_PATH)/bin/hipcc
    GPU_TARGETS ?= $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
    CFLAGS      += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS
    CXXFLAGS    += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS
    LDFLAGS     += -L$(ROCM_PATH)/lib -Wl,-rpath=$(ROCM_PATH)/lib
    LDFLAGS     += -lhipblas -lamdhip64 -lrocblas
    HIPFLAGS    += $(addprefix --offload-arch=,$(GPU_TARGETS))
    WHISPER_OBJ += ggml-cuda.o

ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
    $(HIPCC) $(CXXFLAGS) $(HIPFLAGS) -x hip -c -o $@ $<
endif