| 1 | #!/bin/bash |
| 2 | |
| 3 | # This script build the CPU docker image and run the offline inference inside the container. |
| 4 | # It serves a sanity check for compilation and basic model usage. |
| 5 | set -ex |
| 6 | |
| 7 | # Setup cleanup |
| 8 | remove_docker_container() { docker rm -f cpu-test || true; docker system prune -f; } |
| 9 | trap remove_docker_container EXIT |
| 10 | remove_docker_container |
| 11 | |
| 12 | # Try building the docker image |
| 13 | docker build -t cpu-test -f docker/Dockerfile.s390x . |