ver4a/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

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

Browse source
This commit is contained in:
uvos 2025-02-12 17:25:03 +01:00 • committed by GitHub
parent fef0cbeadf
commit e598697d63
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
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;
} }
} }