[gpu] Exercise new template-generated GPU driver installer

GoogleCloudDataproc · Jan 9, 2025 · 997a61c · 997a61c
1 parent 900c10a
commit 997a61c
Show file tree

Hide file tree

Showing 5 changed files with 334 additions and 1,563 deletions.
diff --git a/gpu/BUILD b/gpu/BUILD
@@ -6,7 +6,11 @@ py_test(
     name = "test_gpu",
     size = "enormous",
     srcs = ["test_gpu.py"],
-    data = ["install_gpu_driver.sh", "mig.sh"],
+    data = [
+      "install_gpu_driver.sh",
+      "verify_pyspark.py",
+      "mig.sh"
+    ],
     local = True,
     shard_count = 15,
     deps = [