| # Copyright 2022 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 |
| |
| steps: |
| - label: ":stopwatch: Benchmark on Nvidia A100 (GCP-a2-highgpu-1g)" |
| key: "run-cuda-benchmark-gcp-gpu-a100" |
| commands: | |
| git clean -fdx |
| buildkite-agent artifact download \ |
| "benchmark-suites-linux-cuda-$${BUILDKITE_BUILD_NUMBER}.tgz" ./ |
| buildkite-agent artifact download \ |
| "iree-linux-tools-$${BUILDKITE_BUILD_NUMBER}.tgz" ./ |
| tar -xzvf "benchmark-suites-linux-cuda-$${BUILDKITE_BUILD_NUMBER}.tgz" |
| tar -xzvf "iree-linux-tools-$${BUILDKITE_BUILD_NUMBER}.tgz" |
| python3 build_tools/benchmarks/run_benchmarks_on_linux.py \ |
| --device_model=GCP-a2-highgpu-1g \ |
| --normal_benchmark_tool_dir=build-linux/tools/ \ |
| --driver_filter_regex=cuda \ |
| -o "benchmark-results-gcp-gpu-a100-$${BUILDKITE_BUILD_NUMBER}.json" \ |
| --verbose \ |
| build-targets/linux-cuda |
| agents: |
| - "gcp:machine-type=a2-highgpu-1g" |
| - "queue=benchmark-cuda" |
| artifact_paths: |
| - "benchmark-results-gcp-gpu-a100-${BUILDKITE_BUILD_NUMBER}.json" |
| timeout_in_minutes: "10" |
