Split cuda tests into separate lib.

PiperOrigin-RevId: 755118711

Author: zkoopmans

test/gpu/BUILD

@@ -84,8 +84,7 @@ go_test(
     visibility = ["//:sandbox"],
     deps = [
         "//pkg/test/dockerutil",
-        "//pkg/test/testutil",
-        "@org_golang_x_sync//errgroup:go_default_library",
+        "//test/gpu/cuda",
     ],
 )
 

test/gpu/cuda/BUILD

@@ -0,0 +1,18 @@
+load("//tools:defs.bzl", "go_library")
+
+package(
+    default_applicable_licenses = ["//:license"],
+    licenses = ["notice"],
+)
+
+go_library(
+    name = "cuda",
+    testonly = True,
+    srcs = ["cuda.go"],
+    visibility = ["//test/gpu:__subpackages__"],
+    deps = [
+        "//pkg/test/dockerutil",
+        "//pkg/test/testutil",
+        "@org_golang_x_sync//errgroup:go_default_library",
+    ],
+)