e2e-code-bench-agent.yaml

#
# Copyright (C) 2019-2023 vdaas.org vald team <vald@vdaas.org>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
name: "Run e2e code bench: bench-agent"
on:
  push:
    branches:
      - main
    paths:
      - ".github/workflows/e2e-bench-agent.yml"
      - "internal/core/**"
      - "internal/client/**"
      - "internal/net/**"
      - "apis/grpc/**"
      - "pkg/agent/core/ngt/**"
      - "cmd/agent/core/ngt/**"
      - "hack/benchmark/**"
      - "versions/GO_VERSION"
      - "versions/NGT_VERSION"
  pull_request:
    paths:
      - ".github/workflows/e2e-bench-agent.yml"
      - "internal/core/**"
      - "internal/client/**"
      - "internal/net/**"
      - "apis/grpc/**"
      - "pkg/agent/core/ngt/**"
      - "cmd/agent/core/ngt/**"
      - "hack/benchmark/**"
      - "versions/GO_VERSION"
      - "versions/NGT_VERSION"

jobs:
  dump_contexts_to_log:
    runs-on: ubuntu-latest
    steps:
      - name: Dump GitHub context
        id: github_context_step
        run: echo $JSON
        env:
          JSON: ${{ toJSON(github) }}
      - name: Dump job context
        run: echo $JSON
        env:
          JSON: ${{ toJSON(job) }}
      - name: Dump steps context
        run: echo $JSON
        env:
          JSON: ${{ toJSON(steps) }}
      - name: Dump runner context
        run: echo $JSON
        env:
          JSON: ${{ toJSON(runner) }}
      - name: Dump strategy context
        run: echo $JSON
        env:
          JSON: ${{ toJSON(strategy) }}
      - name: Dump matrix context
        run: echo $JSON
        env:
          JSON: ${{ toJSON(matrix) }}
  grpc-sequential:
    name: grpc-sequential
    runs-on: ubuntu-latest
    container:
      image: ghcr.io/vdaas/vald/vald-ci-container:nightly
    steps:
      - name: Check out code.
        uses: actions/checkout@v3
        with:
          persist-credentials: false
      - name: set git config
        run: |
          git config --global --add safe.directory ${GITHUB_WORKSPACE}
      - name: Run grpc-sequential
        run: |
          make hack/benchmark/assets/dataset/${DATASET}
          DATASET_ARGS=${DATASET_ARGS} make bench/agent/sequential/grpc
        env:
          DATASET: fashion-mnist-784-euclidean.hdf5
          DATASET_ARGS: fashion-mnist
      - name: Upload artifact
        uses: actions/upload-artifact@v3
        with:
          name: agent-sequential-grpc
          path: |
            ./pprof/agent/sequential/grpc.bin
            ./pprof/agent/sequential/grpc.cpu.out
            ./pprof/agent/sequential/grpc.mem.out
            ./pprof/agent/sequential/grpc.trace.out
            ./pprof/agent/sequential/grpc.cpu.svg
            ./pprof/agent/sequential/grpc.mem.svg

  grpc-stream:
    name: grpc-stream
    runs-on: ubuntu-latest
    container:
      image: ghcr.io/vdaas/vald/vald-ci-container:nightly
    steps:
      - name: Check out code.
        uses: actions/checkout@v3
      - name: set git config
        run: |
          git config --global --add safe.directory ${GITHUB_WORKSPACE}
      - name: Run grpc-stream
        run: |
          make hack/benchmark/assets/dataset/${DATASET}
          DATASET_ARGS=${DATASET_ARGS} make bench/agent/stream
        env:
          DATASET: fashion-mnist-784-euclidean.hdf5
          DATASET_ARGS: fashion-mnist
      - name: Upload artifact
        uses: actions/upload-artifact@v3
        with:
          name: agent-stream
          path: |
            ./pprof/agent/stream.bin
            ./pprof/agent/stream.cpu.out
            ./pprof/agent/stream.mem.out
            ./pprof/agent/stream.trace.out
            ./pprof/agent/stream.cpu.svg
            ./pprof/agent/stream.mem.svg