From 7d75608bbbee84d42dbb1e3dc3390debb9aacf94 Mon Sep 17 00:00:00 2001
From: Huy Do <huydhn@gmail.com>
Date: Tue, 26 Aug 2025 20:32:42 -0700
Subject: [PATCH] Add tests for B200

Signed-off-by: Huy Do <huydhn@gmail.com>
---
 .github/workflows/pr-test.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 71cbd12b9..47408bc65 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -17,6 +17,8 @@ jobs:
             runner: linux.24xlarge
           - config: cuda
             runner: linux.aws.a100
+          - config: cuda
+            runner: linux.dgx.b200
     name: Test ${{ matrix.config }}
     env:
       DOCKER_IMAGE: 'ghcr.io/pytorch/torchbench:latest'