coreflow/test_vx_conformance.sh at main · RunEdgeAI/coreflow · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#! /bin/bash
set -e

TARGET_DIR="${GITHUB_WORKSPACE:-$PWD}"
echo "Running from TARGET_DIR=$TARGET_DIR"

if [ -d "$TARGET_DIR/cts/build" ]; then
  rm -rf $TARGET_DIR/cts/build
fi

mkdir -p $TARGET_DIR/cts/build

# Set the environment variables
export OPENVX_DIR=$TARGET_DIR
export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:$OPENVX_DIR/bazel-bin:$OPENVX_DIR/bazel-bin/vxu:$OPENVX_DIR/bazel-bin/targets/ai_server:$OPENVX_DIR/bazel-bin/targets/c_model:$OPENVX_DIR/bazel-bin/targets/extras:$OPENVX_DIR/bazel-bin/targets/debug:$OPENVX_DIR/bazel-bin/targets/executorch:$OPENVX_DIR/bazel-bin/targets/liteRT:$OPENVX_DIR/bazel-bin/targets/opencl:$OPENVX_DIR/bazel-bin/targets/onnxRT:$OPENVX_DIR/cts/build/lib/
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$OPENVX_DIR/bazel-bin:$OPENVX_DIR/bazel-bin/vxu:$OPENVX_DIR/bazel-bin/targets/ai_server:$OPENVX_DIR/bazel-bin/targets/c_model:$OPENVX_DIR/bazel-bin/targets/extras:$OPENVX_DIR/bazel-bin/targets/debug:$OPENVX_DIR/bazel-bin/targets/executorch:$OPENVX_DIR/bazel-bin/targets/liteRT:$OPENVX_DIR/bazel-bin/targets/opencl:$OPENVX_DIR/bazel-bin/targets/onnxRT:$OPENVX_DIR/cts/build/lib/
export VX_TEST_DATA_PATH=$OPENVX_DIR/cts/test_data/
export VX_CL_SOURCE_DIR=$OPENVX_DIR/kernels/opencl/

# Set the correct library extension based on the OS
if [[ "$(uname)" == "Darwin" ]]; then
  EXT=".dylib"
else
  EXT=".so"
fi

# Build the conformance test suite
cd $TARGET_DIR/cts/build
cmake \
-DCMAKE_POLICY_VERSION_MINIMUM=3.5 \
-DCMAKE_BUILD_TYPE=Debug \
-DOPENVX_INCLUDES=$OPENVX_DIR/include \
-DOPENVX_LIBRARIES="$OPENVX_DIR/bazel-bin/libopenvx$EXT"\;"$OPENVX_DIR/bazel-bin/vxu/libvxu$EXT"\;pthread\;dl\;m \
-DOPENVX_USE_USER_DATA_OBJECT=ON \
-DOPENVX_USE_IX=ON \
-DOPENVX_USE_U1=ON \
-DOPENVX_USE_NN=ON \
-DOPENVX_USE_NN_16=ON \
-DOPENVX_CONFORMANCE_NNEF_IMPORT=ON \
-DOPENVX_CONFORMANCE_NEURAL_NETWORKS=ON \
-DOPENVX_USE_PIPELINING=ON \
-DOPENVX_USE_STREAMING=ON \
..

# -DOPENVX_CONFORMANCE_VISION=ON \
# -DOPENVX_USE_ENHANCED_VISION=ON \

cmake --build .

# Run the conformance test suite
if [[ "$DEBUG" == "1" ]]; then
  echo "Running under GDB since CI is set"
  gdb -batch -ex "run" -ex "bt" --args ./bin/vx_test_conformance
else
  ./bin/vx_test_conformance
fi

# Clean up
cd $TARGET_DIR
rm -rf $TARGET_DIR/cts/build