More updates for review comments

Author: chaxu01

ggml/src/ggml-cpu/CMakeLists.txt

@@ -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()

ggml/src/ggml-cpu/ggml-kleidiai/ggml-kleidiai.cpp

@@ -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];
 }