Skip to content

Commit

Permalink
integration tests for tensorflow
Browse files Browse the repository at this point in the history
  • Loading branch information
uecker committed Jan 19, 2025
1 parent 9c2db5a commit 61741be
Showing 1 changed file with 16 additions and 1 deletion.
17 changes: 16 additions & 1 deletion .gitlab-ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -392,6 +392,7 @@ IntTest:
dependencies:
- Build


IntTest_riscv:
stage: test2
tags:
Expand Down Expand Up @@ -471,6 +472,21 @@ IntTest_MPI_GPU:
dependencies:
- Build_MPI_GPU


IntTest_Tensorflow:
stage: test2
script:
# Already part of the Docker image
# - wget --no-verbose https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.4.0.tar.gz
# - mkdir tensorflow_dir && tar -C tensorflow_dir -xvzf libtensorflow-gpu-linux-x86_64-2.4.0.tar.gz
- python3 -m pip install tensorflow
- AUTOCLEAN=0 TENSORFLOW=1 TENSORFLOW_BASE=/tensorflow_dir/ CUDA=1 CUDA_LIB=lib64 GPUARCH_FLAGS="-arch sm_35 -Wno-deprecated-gpu-targets" make test
needs: [Build_Tensorflow]
dependencies:
- Build_Tensorflow
allow_failure: true


#IntTest_LTO:
# stage: test2
# script:
Expand Down Expand Up @@ -600,7 +616,6 @@ Ubuntu_24.04:
- make all -j8
- make utest -j8


Arch_Latest:
variables:
DEBUG: "0"
Expand Down

0 comments on commit 61741be

Please sign in to comment.