| # Copyright 2021 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| # An image for running `convperf`: https://github.com/nod-ai/convperf. |
| # |
| # `convperf` benchmarks convolution workloads on IREE and other backends such |
| # as libxsmm. IREE is included as a submodule. |
| |
| FROM gcr.io/iree-oss/perf@sha256:8f4dfc8cf779f6499215a5c4ff2c7608ac3329f23f6ddcf6025e1868e1584222 |
| |
| ######## OpenMP ######## |
| RUN apt-get update \ |
| && apt-get install -y libomp-14-dev |
| ############## |
| |
| ######## ConvPerf ######## |
| COPY build_tools/docker/context/setup_convperf.sh /usr/local/bin |
| |
| ARG CONVPERF_SHA="ca181bf9f1b959477acf8043eb5b5579fc47a37f" |
| |
| # Generate a version of mmperf for CPU. |
| RUN mkdir -p "/usr/local/src/convperf" \ |
| && /usr/local/bin/setup_convperf.sh "/usr/local/src/convperf" "${CONVPERF_SHA}" |
| |
| ENV CONVPERF_REPO_DIR="/usr/local/src/convperf/convperf" |
| ############## \ |
