File tree Expand file tree Collapse file tree 3 files changed +14
-3
lines changed Expand file tree Collapse file tree 3 files changed +14
-3
lines changed Original file line number Diff line number Diff line change @@ -245,7 +245,7 @@ jobs:
245
245
toolchain_cmake=$(realpath ${toolchain_cmake})
246
246
247
247
# Build and test size test
248
- bash test/build_size_test.sh "-DCMAKE_TOOLCHAIN_FILE=${toolchain_cmake} -DEXECUTORCH_BUILD_ARM_BAREMETAL=ON"
248
+ bash test/build_size_test.sh "-DCMAKE_TOOLCHAIN_FILE=${toolchain_cmake} -DEXECUTORCH_BUILD_ARM_BAREMETAL=ON -DEXECUTORCH_BUILD_NXP_NEUTRON=ON "
249
249
elf="cmake-out/test/size_test"
250
250
251
251
# Dump basic info
@@ -271,6 +271,15 @@ jobs:
271
271
exit 1
272
272
fi
273
273
274
+ # Build check for the neutron backend library
275
+ lib_neutron = "cmake-out/backends/nxp/libexecutorch_delegate_neutron.a"
276
+ if [ ! -f $(lib_neutron) ]; then
277
+ echo "Neutron backend library not found!"
278
+ exit 1
279
+ else
280
+ echo "Neutron backend library built."
281
+ fi
282
+
274
283
test-coreml-delegate :
275
284
name : test-coreml-delegate
276
285
uses : pytorch/test-infra/.github/workflows/macos_job.yml@main
Original file line number Diff line number Diff line change 6
6
set (
7
7
_common_include_directories
8
8
${CMAKE_CURRENT_SOURCE_DIR} /../../..
9
+ ${CMAKE_CURRENT_SOURCE_DIR} /../../runtime/core/portable_type/c10
9
10
)
11
+ add_compile_definitions (C10_USING_CUSTOM_GENERATED_MACROS )
10
12
11
13
set (_neutron_sources ${CMAKE_CURRENT_SOURCE_DIR} /runtime/NeutronBackend.cpp )
12
14
Original file line number Diff line number Diff line change @@ -236,8 +236,8 @@ class NeutronBackend final : public PyTorchBackendInterface {
236
236
// cfg->mcfg.kernels
237
237
const uint8_t * transpositionFlags =
238
238
static_cast <const uint8_t *>(processed->data ());
239
- uint32_t numInputs = transpositionFlags[INPUT_TENSOR_FORMAT_LEN_POS];
240
- uint32_t numOutputs = transpositionFlags[OUTPUT_TENSOR_FORMAT_LEN_POS];
239
+ int numInputs = transpositionFlags[INPUT_TENSOR_FORMAT_LEN_POS];
240
+ int numOutputs = transpositionFlags[OUTPUT_TENSOR_FORMAT_LEN_POS];
241
241
cfg->inputTranspositionFlags =
242
242
INPUT_TENSOR_FORMAT_ARRAY_ADDR (transpositionFlags);
243
243
cfg->outputTranspositionFlags =
You can’t perform that action at this time.
0 commit comments