From 9425ba247813e52774fbbdb282b7e1704e21c34a Mon Sep 17 00:00:00 2001
From: Cheng Wan <54331508+ch-wan@users.noreply.github.com>
Date: Wed, 10 Jun 2026 21:16:14 -0700
Subject: [PATCH] Remove extra_pytest_path from pr-test.yml (#27856)

---
 .github/workflows/pr-test.yml                 | 1 -
 test/registered/jit/test_flash_attention_4.py | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 1887be7a3..0f183b20c 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -388,7 +388,6 @@ jobs:
       caller_inputs: ${{ toJson(inputs) }}
       partitions: ${{ needs.check-changes.outputs.partitions }}
       run_timeout_minutes: '40'
-      extra_pytest_path: 'test/registered/jit/test_flash_attention_4.py'
     secrets: inherit
 
   call-multimodal-gen-tests:
diff --git a/test/registered/jit/test_flash_attention_4.py b/test/registered/jit/test_flash_attention_4.py
index 95bb978af..a3d45b6bf 100644
--- a/test/registered/jit/test_flash_attention_4.py
+++ b/test/registered/jit/test_flash_attention_4.py
@@ -15,6 +15,7 @@ from sglang.jit_kernel.flash_attention import flash_attn_varlen_func
 from sglang.test.ci.ci_register import register_cuda_ci
 
 register_cuda_ci(est_time=120, suite="base-b-kernel-unit-1-gpu-large")
+register_cuda_ci(est_time=120, suite="base-b-kernel-unit-1-gpu-b200")
 register_cuda_ci(est_time=900, suite="nightly-kernel-1-gpu", nightly=True)
 
 # Skip this test on Hopper machine