Add GPU benchmark comment (#688)

argumentcomputer · Sep 20, 2023 · fe65ab4 · fe65ab4
1 parent 078ceaa
commit fe65ab4
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 6 deletions.
diff --git a/.github/workflows/bench_pr_comment.yml b/.github/workflows/bench_pr_comment.yml
@@ -12,15 +12,14 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  runBenchmark:
-    name: run benchmark
+  run-benchmark:
+    name: run end2end benchmark
     runs-on: [self-hosted, bench]
     if:
       github.event.issue.pull_request
       && github.event.issue.state == 'open'
       && contains(github.event.comment.body, '!benchmark')
-      && (github.event.comment.author_association == 'MEMBER'
-      || github.event.comment.author_association == 'OWNER')
+      && (github.event.comment.author_association == 'MEMBER' || github.event.comment.author_association == 'OWNER')
     steps:
       - uses: xt0rted/pull-request-comment-branch@v2
         id: comment-branch
@@ -38,3 +37,29 @@ jobs:
           benchName: "end2end"
           # Needed. The name of the branch to compare with
           branchName: ${{ github.ref_name }}
+
+  gpu-benchmark:
+    name: run fibonacci benchmark on GPU
+    runs-on: [self-hosted, gpu-bench]
+    if:
+      github.event.issue.pull_request
+      && github.event.issue.state == 'open'
+      && contains(github.event.comment.body, '!benchmark')
+      && (github.event.comment.author_association == 'MEMBER' || github.event.comment.author_association == 'OWNER')
+    steps:
+      - uses: xt0rted/pull-request-comment-branch@v2
+        id: comment-branch
+
+      - uses: actions/checkout@v4
+        if: success()
+        with:
+          ref: ${{ steps.comment-branch.outputs.head_ref }}
+      # Set the Rust env vars
+      - uses: actions-rs/toolchain@v1
+      - uses: Swatinem/rust-cache@v2
+      - uses: boa-dev/criterion-compare-action@v3
+        with:
+          # Optional. Compare only this benchmark target
+          benchName: "fibonacci"
+          # Needed. The name of the branch to compare with
+          branchName: ${{ github.ref_name }}
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -38,8 +38,6 @@ jobs:
     env:
       RUSTFLAGS: -D warnings
     steps:
-      - name: Install dependencies
-        run: sudo apt-get install -y pkg-config libssl-dev
       - uses: actions/checkout@v4
         with:
           submodules: recursive