ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-03-01 03:26:12 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

HIP: Switch to std::vector in rocblas version check (llama/11820)

Browse Source
This commit is contained in:
uvos 2025-02-12 17:25:03 +01:00 committed by Georgi Gerganov
parent 7fbca6304e
commit 86729fcd6d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ggml/src/ggml-cuda/ggml-cuda.cu
View File

@ -178,11 +178,11 @@ static ggml_cuda_device_info ggml_cuda_init() {
int major_version = 0; int major_version = 0;
size_t version_length = 0; size_t version_length = 0;
if (rocblas_get_version_string_size(&version_length) == rocblas_status_success) { if (rocblas_get_version_string_size(&version_length) == rocblas_status_success) {
std::string version(version_length, '\0'); std::vector<char> version(version_length+1, '\0');
if (rocblas_get_version_string(version.data(), version.size()) == rocblas_status_success) { if (rocblas_get_version_string(version.data(), version.size()) == rocblas_status_success) {
version.resize(::strlen(version.c_str())); version.resize(::strlen(version.data()));
int parsed_value = 0; int parsed_value = 0;
if (std::from_chars(version.c_str(), version.c_str() + version.length(), parsed_value).ec == std::errc()) { if (std::from_chars(version.data(), version.data() + version.size(), parsed_value).ec == std::errc()) {
major_version = parsed_value; major_version = parsed_value;
} }
} }