Update Tensorflow to 2.16.1 #2424
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Tests | |
on: | |
push: | |
branches: | |
- main | |
pull_request: {} | |
concurrency: | |
group: ${{ github.workflow }}-${{ github.ref }} | |
cancel-in-progress: true | |
env: | |
TF_PYTHON_VERSION: "3.11" | |
jobs: | |
TFLite: | |
runs-on: ubuntu-latest | |
if: "!contains(github.event.head_commit.message, 'ci-skip')" | |
steps: | |
- uses: actions/checkout@v4 | |
with: | |
submodules: true | |
fetch-depth: 0 | |
- name: Run C++ Unit Tests | |
run: bazelisk test larq_compute_engine/tests:cc_tests --test_output=all | |
- name: Build TF Lite Static Library with CMake | |
run: | | |
mkdir build | |
cd build | |
cmake .. | |
make -j2 | |
ARM: | |
runs-on: ubuntu-latest | |
if: "!contains(github.event.head_commit.message, 'ci-skip')" | |
steps: | |
- uses: actions/checkout@v4 | |
- name: Install qemu-user | |
run: | | |
sudo apt-get update | |
sudo apt-get install -y --no-install-recommends qemu-user | |
- name: "TF Lite Arm32: Cross-compile and run unit tests in qemu" | |
run: bazelisk test larq_compute_engine/tests:arm32_tests --config=rpi3 --test_output=all --test_filter="-*BigTest*" --copt=-O1 | |
- name: "TF Lite Aarch64: Cross-compile and run unit tests in qemu" | |
run: bazelisk test larq_compute_engine/tests:aarch64_tests --config=aarch64 --test_output=all --test_filter="-*BigTest*" | |
- name: "Benchmark utility: check it builds successfully" | |
run: bazelisk build //larq_compute_engine/tflite/benchmark:lce_benchmark_model --config=aarch64 --compilation_mode=fastbuild | |
MLIR: | |
runs-on: ubuntu-latest | |
if: "!contains(github.event.head_commit.message, 'ci-skip')" | |
steps: | |
- uses: actions/checkout@v4 | |
- uses: google-github-actions/auth@v2 | |
continue-on-error: true | |
with: | |
credentials_json: ${{ secrets.gcs_bazel_cache }} | |
- name: Configure Bazel | |
run: | | |
if [[ -n $GOOGLE_APPLICATION_CREDENTIALS ]]; then | |
echo -e 'build --remote_cache=https://storage.googleapis.com/plumerai-bazel-cache/lce-ubuntu' >> .bazelrc.user | |
echo -e 'build --google_default_credentials' >> .bazelrc.user | |
fi | |
shell: bash | |
- name: Run Interpreter test | |
run: bazelisk test larq_compute_engine/tflite/tests:interpreter_test --test_output=all | |
- name: Run FileCheck tests | |
run: bazelisk test larq_compute_engine/mlir/tests:all --test_output=all | |
- name: Run End2End tests | |
run: bazelisk test larq_compute_engine/tests:end2end_test --test_output=all --test_env=TF_USE_LEGACY_KERAS=1 | |
- name: Run Strip dequantize op tests | |
run: bazelisk test larq_compute_engine/tests:strip_lcedequantize_test --test_output=all --test_env=TF_USE_LEGACY_KERAS=1 | |
ConverterPython: | |
runs-on: ubuntu-latest | |
strategy: | |
matrix: | |
tf-version: [2.5.3, 2.6.5, 2.7.4, 2.8.4, 2.9.3, 2.10.1, 2.11.0] | |
python-version: [3.9] | |
flatbuffers-version: [2.0] | |
protobuf-version: [3.19.6] | |
include: | |
- tf-version: 2.13.0 | |
python-version: 3.11 | |
flatbuffers-version: 23.1.21 | |
protobuf-version: 4.23.4 | |
- tf-version: 2.16.1 | |
python-version: 3.11 | |
flatbuffers-version: 24.3.25 | |
protobuf-version: 4.25.3 | |
if: "!contains(github.event.head_commit.message, 'ci-skip')" | |
steps: | |
- uses: actions/checkout@v4 | |
- uses: actions/setup-python@v5 | |
with: | |
python-version: ${{ matrix.python-version }} | |
- name: Install TensorFlow | |
run: pip install tensorflow==${{matrix.tf-version}} --no-cache-dir | |
- name: Install legacy tf-keras | |
if: matrix.tf-version == '2.16.1' | |
run: pip install tf-keras==2.16.0 | |
- name: Install flatbuffers | |
run: pip install flatbuffers==${{matrix.flatbuffers-version}} --no-cache-dir | |
- name: Install protobuf | |
run: pip install protobuf==${{matrix.protobuf-version}} --no-cache-dir | |
- name: Install other dependencies | |
run: pip install larq~=0.13.3 packaging tqdm --no-cache-dir | |
- name: Run Converter test | |
run: TF_USE_LEGACY_KERAS=1 PYTHONPATH=./ python larq_compute_engine/mlir/python/converter_test.py | |
Android_AAR: | |
runs-on: ubuntu-latest | |
if: "!contains(github.event.head_commit.message, 'ci-skip')" | |
steps: | |
- uses: actions/checkout@v4 | |
with: | |
fetch-depth: 0 | |
- uses: actions/setup-python@v5 | |
with: | |
python-version: 3.9 | |
- uses: actions/cache@v4 | |
id: cache | |
with: | |
path: /tmp/lce_android | |
key: ${{ runner.os }}-${{ hashFiles('**/third_party/install_android.sh') }} | |
- name: Set Java version | |
run: echo "JAVA_HOME=${JAVA_HOME_8_X64}" >> $GITHUB_ENV | |
- name: Download and install Android NDK/SDK | |
if: steps.cache.outputs.cache-hit != 'true' | |
run: ./third_party/install_android.sh | |
- name: Configure Bazel | |
run: LCE_SET_ANDROID_WORKSPACE=1 ANDROID_SDK_HOME="/tmp/lce_android" ANDROID_API_LEVEL=30 ANDROID_NDK_HOME="/tmp/lce_android/ndk/25.2.9519653" ANDROID_NDK_API_LEVEL=30 ANDROID_BUILD_TOOLS_VERSION=31.0.0 ./configure.py | |
shell: bash | |
- name: Build LCE AAR | |
run: BUILDER=bazelisk ./larq_compute_engine/tflite/java/build_lce_aar.sh |