More updates for review comments

This commit is contained in:
Charles Xu 2025-02-07 17:42:54 +01:00
parent 9edd10737b
commit e04880fa54
2 changed files with 2 additions and 2 deletions

View file

@ -116,7 +116,7 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
set(ARM_MCPU_FLAG_FIX "${ARM_MCPU_FLAG_FIX}+${tag}" PARENT_SCOPE)
else()
set(CMAKE_REQUIRED_FLAGS "${ARM_MCPU_FLAG}+no${tag}")
check_cxx_source_compiles("${code}" GGML_MACHINE_SUPPORTS_no${tag})
check_cxx_source_compiles("int main() { return 0; }" GGML_MACHINE_SUPPORTS_no${tag})
if (GGML_MACHINE_SUPPORTS_no${tag})
set(ARM_MCPU_FLAG_FIX "${ARM_MCPU_FLAG_FIX}+no${tag}" PARENT_SCOPE)
endif()

View file

@ -61,7 +61,7 @@ static void init_kleidiai_context(void) {
ggml_critical_section_end();
}
static inline int ggml_ne(const ggml_tensor * tensor, int dim) {
static inline int64_t ggml_ne(const ggml_tensor * tensor, int dim) {
GGML_ASSERT(dim >= 0 && dim < GGML_MAX_DIMS);
return tensor->ne[dim];
}