ane test

metascroy · Dec 19, 2024 · e648cec · e648cec
1 parent f341da8
commit e648cec
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.ci/scripts/gather_benchmark_configs.py b/.ci/scripts/gather_benchmark_configs.py
@@ -41,6 +41,7 @@
         "mps",
         # TODO: Add support for llama3 ane
         # "llama3_coreml_ane",
+        "llama3_coreml_ane",
     ],
 }
 

diff --git a/.github/workflows/apple-perf.yml b/.github/workflows/apple-perf.yml
@@ -261,7 +261,7 @@ jobs:
                 -kv \
                 --disable_dynamic_shape \
                 --coreml \
-                --coreml-ios 17 \
+                --coreml-ios 18 \
                 --coreml-quantize c4w \
                 --coreml-compute-units cpu_and_ne \
                 --output_name="${OUT_ET_MODEL_NAME}.pte"

diff --git a/build/build_apple_frameworks.sh b/build/build_apple_frameworks.sh
@@ -24,7 +24,7 @@ HEADERS_PATH="include"
 
 PLATFORMS=("ios" "simulator" "macos")
 PLATFORM_FLAGS=("OS64" "SIMULATORARM64" "MAC_ARM64")
-PLATFORM_TARGET=("17.0" "17.0" "10.15")
+PLATFORM_TARGET=("18.0" "18.0" "10.15")
 
 FRAMEWORK_EXECUTORCH="executorch:\
 libexecutorch.a,\
-Original file line number
+Diff line change
@@ Expand Up / @@ -41,6 +41,7 @@ @@
             "mps",
             # TODO: Add support for llama3 ane
             # "llama3_coreml_ane",
+            "llama3_coreml_ane",
         ],
     }
@@ Expand Down @@