|
|
# include "ggml-backend-impl.h" |
|
|
|
|
|
#if defined(__powerpc64__) || defined(__ppc64__) || defined(__PPC64__) |
|
|
|
|
|
#if defined(__linux__) |
|
|
#include <sys/auxv.h> |
|
|
#endif |
|
|
|
|
|
#include <string> |
|
|
|
|
|
struct powerpc_features { |
|
|
std::string platform = ""; |
|
|
int power_version = -1; |
|
|
|
|
|
bool has_vsx = false; |
|
|
|
|
|
powerpc_features() { |
|
|
#if defined(__linux__) |
|
|
unsigned long auxval = getauxval(AT_PLATFORM); |
|
|
if (auxval) { |
|
|
platform = std::string(reinterpret_cast<const char*>(auxval)); |
|
|
|
|
|
if (platform.substr(0, 5) == "power") { |
|
|
|
|
|
int vpos = -1; |
|
|
for (int i = platform.length() - 1; i >= 0; i--) { |
|
|
if (std::isdigit(platform[i])) { |
|
|
vpos = i; |
|
|
} else { |
|
|
break; |
|
|
} |
|
|
} |
|
|
if (vpos > -1) { |
|
|
power_version = std::stoi(platform.substr(vpos)); |
|
|
} |
|
|
} |
|
|
} |
|
|
#endif |
|
|
if (power_version >= 9) { |
|
|
has_vsx = true; |
|
|
} |
|
|
} |
|
|
}; |
|
|
|
|
|
static int ggml_backend_cpu_powerpc_score() { |
|
|
int score = 1; |
|
|
powerpc_features pf; |
|
|
|
|
|
|
|
|
#if defined(GGML_USE_POWER7) |
|
|
if (pf.power_version < 7) { return 0; } |
|
|
score += 1<<1; |
|
|
#endif |
|
|
#if defined(GGML_USE_POWER8) |
|
|
if (pf.power_version < 8) { return 0; } |
|
|
score += 1<<2; |
|
|
#endif |
|
|
#if defined(GGML_USE_POWER9) |
|
|
if (pf.power_version < 9) { return 0; } |
|
|
score += 1<<3; |
|
|
#endif |
|
|
#if defined(GGML_USE_POWER10) |
|
|
if (pf.power_version < 10) { return 0; } |
|
|
score += 1<<4; |
|
|
#endif |
|
|
#if defined(GGML_USE_POWER11) |
|
|
if (pf.power_version < 11) { return 0; } |
|
|
score += 1<<5; |
|
|
#endif |
|
|
|
|
|
|
|
|
#if defined(GGML_USE_VSX) |
|
|
if (!pf.has_vsx) { return 0; } |
|
|
score += 1<<6; |
|
|
#endif |
|
|
|
|
|
return score; |
|
|
} |
|
|
|
|
|
GGML_BACKEND_DL_SCORE_IMPL(ggml_backend_cpu_powerpc_score) |
|
|
|
|
|
#endif |
|
|
|
