Skip to content

Turn on Compute library testing in CI for AArch64 #8291

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mbrookhart merged 3 commits into from
Jun 22, 2021
Merged

Turn on Compute library testing in CI for AArch64 #8291

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
0f6aac7
Turn on Compute library testing in CI.
Jun 18, 2021
f97f773
Fix typo
Jun 21, 2021
dca09cd
Fix up use of ubuntu_install_arm_compute_lib.sh in Dockerfile.ci_arm
Jun 22, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions docker/Dockerfile.ci_arm
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,5 +43,5 @@ COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
RUN bash /install/ubuntu_install_redis.sh

# Arm(R) Compute Library
COPY install/ubuntu_install_arm_compute_lib.sh /install/ubuntu_install_arm_compute_lib.sh
RUN bash /install/ubuntu_install_arm_compute_lib.sh
COPY install/ubuntu_download_arm_compute_lib_binaries.sh /install/ubuntu_download_arm_compute_lib_binaries.sh
RUN bash /install/ubuntu_download_arm_compute_lib_binaries.sh
3 changes: 2 additions & 1 deletion docker/install/ubuntu_download_arm_compute_lib_binaries.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ target_lib="linux-arm64-v8a-neon"
# target_lib="${target_lib}-asserts"

extract_dir="arm_compute-${compute_lib_version}-bin-linux"
install_path="/opt/arm/acl"
install_path="/opt/acl"

tmpdir=$(mktemp -d)

Expand All @@ -54,6 +54,7 @@ cd "$tmpdir"
curl -sL "${compute_lib_download_url}" -o "${compute_lib_file_name}"
tar xzf "${compute_lib_file_name}"

rm -rf "${install_path}"
mkdir -p "${install_path}"
cp -r "${extract_dir}/include" "${install_path}/"
cp -r "${extract_dir}/arm_compute" "${install_path}/include/"
Expand Down
25 changes: 20 additions & 5 deletions tests/python/contrib/test_arm_compute_lib/test_network.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,10 @@ def _build_and_run_network(mod, params, inputs, device, tvm_ops, acl_partitions,

def _get_tflite_model(tflite_model_path, inputs_dict):
"""Convert TFlite graph to relay."""
import tflite.Model
try:
import tflite.Model
except ImportError:
pytest.skip("Missing Tflite support")

with open(tflite_model_path, "rb") as f:
tflite_model_buffer = f.read()
Expand Down Expand Up @@ -92,7 +95,10 @@ def test_vgg16():
device = Device()

def get_model():
from keras.applications import VGG16
try:
from keras.applications import VGG16
except ImportError:
pytest.skip("Missing Keras Package")

vgg16 = VGG16(include_top=True, weights="imagenet", input_shape=(224, 224, 3), classes=1000)
inputs = {vgg16.input_names[0]: ((1, 224, 224, 3), "float32")}
Expand All @@ -113,7 +119,10 @@ def test_mobilenet():
device = Device()

def get_model():
from keras.applications import MobileNet
try:
from keras.applications import MobileNet
except ImportError:
pytest.skip("Missing keras module")

mobilenet = MobileNet(
include_top=True, weights="imagenet", input_shape=(224, 224, 3), classes=1000
Expand All @@ -133,7 +142,10 @@ def test_quantized_mobilenet():
if skip_runtime_test():
return

import tvm.relay.testing.tf as tf_testing
try:
import tvm.relay.testing.tf as tf_testing
except ImportError:
pytest.skip("Missing Tflite support")

device = Device()

Expand All @@ -158,7 +170,10 @@ def test_squeezenet():
if skip_runtime_test():
return

import tvm.relay.testing.tf as tf_testing
try:
import tvm.relay.testing.tf as tf_testing
except ImportError:
pytest.skip("Missing TF Support")
Copy link
Contributor

@lhutton1 lhutton1 Jun 22, 2021

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
pytest.skip("Missing TF Support")
pytest.skip("Missing Tflite Support")


device = Device()

Expand Down
1 change: 1 addition & 0 deletions tests/scripts/task_config_build_arm.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,3 +34,4 @@ echo set\(CMAKE_CXX_FLAGS -Werror\) >> config.cmake
echo set\(USE_VTA_TSIM ON\) >> config.cmake
echo set\(USE_VTA_FSIM ON\) >> config.cmake
echo set\(USE_ARM_COMPUTE_LIB ON\) >> config.cmake
echo set\(USE_ARM_COMPUTE_LIB_GRAPH_EXECUTOR "/opt/acl"\) >> config.cmake