More updates for review comments
This commit is contained in:
parent
9edd10737b
commit
e04880fa54
2 changed files with 2 additions and 2 deletions
|
@ -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()
|
||||
|
|
|
@ -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];
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue