diff --git a/.github/configs/nvidia-master.yaml b/.github/configs/nvidia-master.yaml index e08778cc5..4f8ea4ca7 100644 --- a/.github/configs/nvidia-master.yaml +++ b/.github/configs/nvidia-master.yaml @@ -170,7 +170,7 @@ gptoss-fp4-b200-trt: image: nvcr.io#nvidia/tensorrt-llm/release:1.2.0rc0.post1 model: openai/gpt-oss-120b model-prefix: gptoss - runner: b200-nvs + runner: b200-trt precision: fp4 framework: trt # For all sequence lengths, if CONC >= 256, then EP=TP and DP_ATTN=true diff --git a/.github/configs/runners.yaml b/.github/configs/runners.yaml index cdd865561..ccbb15cbc 100644 --- a/.github/configs/runners.yaml +++ b/.github/configs/runners.yaml @@ -28,12 +28,9 @@ h200-trt: b200-trt: - 'b200-nv_0' - 'b200-nv_1' -b200-nvs: -- 'b200-nv_0' -- 'b200-nv_1' -b200: - 'b200-nb_0' - 'b200-nb_1' +b200: - 'b200-nvd_0' - 'b200-nvd_1' - 'b200-nvd_2'