From 3574bdd750349e806433ee328e6f7cc17d8c6776 Mon Sep 17 00:00:00 2001
From: Zach Koopmans <zkoopmans@google.com>
Date: Wed, 25 Oct 2023 21:53:34 -0700
Subject: [PATCH] Make GPU tests on presubmit use only latest driver.

PiperOrigin-RevId: 576738146
---
 .buildkite/pipeline.yaml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/.buildkite/pipeline.yaml b/.buildkite/pipeline.yaml
index ed6d7889a..044bc0354 100644
--- a/.buildkite/pipeline.yaml
+++ b/.buildkite/pipeline.yaml
@@ -169,11 +169,20 @@ steps:
       - GOOS=linux GOARCH=mips go build $$PACKAGES
 
   # GPU workflow.
+  - <<: *common
+    <<: *source_test
+    label: ":firefighter: GPU Smoke Test"
+    commands:
+      - make sudo TARGETS=//tools/gpu:main ARGS="install --latest" || cat /var/log/nvidia-installer.log
+      - make gpu-tests
+    agents:
+      queue: gpu
   - <<: *common
     <<: *source_test
     label: ":screwdriver: All GPU Drivers Test"
     commands:
       - tools/gpu/all_drivers_test.sh
+    if: build.branch == "master"
     agents:
       queue: gpu
   # Release workflow.