diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index ce199ed0eee85b1b11a28054949ed7f935a9f6ee..e10a2c63baea28b33cb47842d191044b740b147b 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -155,7 +155,8 @@ dynamic_linking_test: DEVICE_CONF_FILE=generic-mobile-devices/devices.yml fi - if [ -z "$TARGET_SOCS" ]; then TARGET_SOCS=random; fi - - python tools/converter.py convert --config=${CONF_FILE} --target_socs=$TARGET_SOCS --model_graph_format=file --model_data_format=file --cl_mem_type=buffer + - python tools/converter.py convert --config=${CONF_FILE} --target_socs=$TARGET_SOCS --model_graph_format=file --model_data_format=file --cl_mem_type=buffer || exit 1; + - python tools/converter.py run --config=${CONF_FILE} --target_socs=$TARGET_SOCS --device_yml=${DEVICE_CONF_FILE} --round=1 --validate --model_graph_format=file --model_data_format=file --mace_lib_type=dynamic || exit 1; - rm -rf mace-models only: - triggers diff --git a/mace/tools/BUILD.bazel b/mace/tools/BUILD.bazel index 95d1f38662415e68f1a2a5898ac1081a175339db..df620200b2c108388a6ff6f005f9c4149986796e 100644 --- a/mace/tools/BUILD.bazel +++ b/mace/tools/BUILD.bazel @@ -49,7 +49,6 @@ cc_binary( ]), linkstatic = 0, deps = [ - ":statistics", "//external:gflags_nothreads", "//mace/codegen:generated_mace_engine_factory", "//mace/codegen:generated_models", diff --git a/tools/python/encrypt.py b/tools/python/encrypt.py index 1ea78d965505e94d936347f760e1c36e3b94af61..79440f4569e72a42bee047625394553657696ed7 100644 --- a/tools/python/encrypt.py +++ b/tools/python/encrypt.py @@ -134,7 +134,7 @@ def save_model_to_code(namespace, model, params, model_checksum, try: device = cvt.DeviceType[device.upper()] except: # noqa - if device.upper == "DSP": + if device.upper() == "DSP": device = cvt.DeviceType.HEXAGON else: device = cvt.DeviceType.CPU @@ -145,7 +145,7 @@ def save_model_to_code(namespace, model, params, model_checksum, end=min(start + 10, op_size), net=model, tag=namespace, - device=device, + device=device.value, ) with open(output + "/op" + str(counter) + ".cc", "w") as f: f.write(source)