File tree Expand file tree Collapse file tree 2 files changed +10
-5
lines changed
backends/apple/coreml/test Expand file tree Collapse file tree 2 files changed +10
-5
lines changed Original file line number Diff line number Diff line change 4
4
5
5
import unittest
6
6
7
+ import executorch .exir
8
+
7
9
import torch
8
10
import torchvision
9
- import executorch .exir
10
11
11
- from executorch .backends .apple .coreml .compiler import CoreMLBackend
12
- from executorch .backends .apple .coreml .partition .coreml_partitioner import CoreMLPartitioner
12
+ from executorch .backends .apple .coreml .partition .coreml_partitioner import (
13
+ CoreMLPartitioner ,
14
+ )
13
15
14
16
15
17
class TestCoreMLPartitioner (unittest .TestCase ):
@@ -54,7 +56,9 @@ def test_vit_skip_conv(self):
54
56
exir_program_aten = torch .export .export (model , example_inputs )
55
57
edge_program_manager = executorch .exir .to_edge (exir_program_aten )
56
58
delegated_program_manager = edge_program_manager .to_backend (
57
- CoreMLPartitioner (skip_ops_for_coreml_delegation = ["aten.convolution.default" ])
59
+ CoreMLPartitioner (
60
+ skip_ops_for_coreml_delegation = ["aten.convolution.default" ]
61
+ )
58
62
)
59
63
60
64
assert [
Original file line number Diff line number Diff line change @@ -20,7 +20,8 @@ add_subdirectory(
20
20
21
21
if (NOT CMAKE_TOOLCHAIN_FILE MATCHES ".*ios\. toolchain\. cmake$" )
22
22
add_library (executorch_jni SHARED jni/jni_layer.cpp )
23
- target_link_libraries (executorch_jni extension_data_loader extension_module xnn_executor_runner_lib fbjni )
23
+ target_link_libraries (executorch_jni extension_data_loader
24
+ extension_module xnn_executor_runner_lib fbjni )
24
25
if (EXECUTORCH_BUILD_QNN )
25
26
target_link_libraries (executorch_jni qnn_executorch_backend )
26
27
endif ()
You can’t perform that action at this time.
0 commit comments