补充

Tencent · Sep 13, 2024 · b0ed35d · b0ed35d
1 parent 591b55d
commit b0ed35d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/cpu.cpp b/src/cpu.cpp
@@ -2160,6 +2160,12 @@ int cpu_support_arm_vfpv4()
 int cpu_support_arm_asimdhp()
 {
     try_initialize_global_cpu_info();
+    // ¼ì²é»·¾³±äÁ¿ NCNN_ISA
+    const char* ncnnIsaEnv = std::getenv("NCNN_ISA");
+    if (ncnnIsaEnv && strstr(ncnnIsaEnv, "+asimdhp") != nullptr)
+    {
+        return true;
+    }
 #if __aarch64__
 #if defined _WIN32
     return g_cpu_support_arm_asimdhp;