feat: ROCm linux support - Githubissues

tazz4843 / whisper-rs

Rust bindings to https://github.com/ggerganov/whisper.cpp

The Unlicense

607 stars 105 forks source link

feat: ROCm linux support #154

Closed newfla closed 3 weeks ago

newfla commented 4 weeks ago

The PR introduces the support for building the library with rocm/hipblas support. The following changes have been made:

Added hipblas feature flag to whisper-rs and whisper-rs-sys
ROCm install folder can be optionally specified by env::HIP_PATH. Defaults to /opt/rocm
Gpu targets can be optionally specified by env::AMDGPU_TARGETS
Panics when targeting windows

Unfortunately I wasn't able to build whisper.cpp on windows: generated vcxproj refuse to use hipcc and defaults to cl.exe 😣

tazz4843 commented 3 weeks ago

Thanks :)