Skip to content

Commit 6bb948b

Browse files
committed
[gpu] Exercise new template-generated GPU driver installer
gpu/test_gpu.py: * using tests from #1275 gpu/verify_pyspark.py: * new test file ; will probably be moved to mlvm templates/gpu/install_gpu_driver.sh.in: * this action template includes only the code unique to this action
1 parent 824bcf8 commit 6bb948b

File tree

4 files changed

+334
-94
lines changed

4 files changed

+334
-94
lines changed

gpu/BUILD

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,11 @@ py_test(
66
name = "test_gpu",
77
size = "enormous",
88
srcs = ["test_gpu.py"],
9-
data = ["install_gpu_driver.sh", "mig.sh"],
9+
data = [
10+
"install_gpu_driver.sh",
11+
"verify_pyspark.py",
12+
"mig.sh"
13+
],
1014
local = True,
1115
shard_count = 15,
1216
deps = [

0 commit comments

Comments
 (0)