Skip to content

Commit 5d72669

Browse files
zkoopmansgvisor-bot
authored andcommitted
Add more CUDA 12.8 tests
Add two more suites of tests to the CUDA 12.8 tests after testing on two types of GPUs. Also add a test definition in the release pipeline. PiperOrigin-RevId: 758393605
1 parent 65b2cb8 commit 5d72669

File tree

2 files changed

+22
-33
lines changed

2 files changed

+22
-33
lines changed

.buildkite/release.yaml

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ steps:
7676
- make gpu-all-tests
7777
agents:
7878
queue: gpu
79-
- label: ":fish: CUDA tests"
79+
- label: ":fish: CUDA 12.2 tests"
8080
# This is its own test rather than being part of the GPU tests,
8181
# because it takes around 30 minutes to run.
8282
parallelism: 32
@@ -88,6 +88,19 @@ steps:
8888
- make cuda-tests ARGS="--cuda_verify_compatibility=true"
8989
agents:
9090
queue: gpu
91+
- label: ":batfish: CUDA 12.8 tests"
92+
if: build.env("STAGED_BINARIES") == null && ( build.env("SKIP_GPU_TESTS") == null || build.message =~ /nvidia|nvproxy|gpu/i )
93+
# This is its own test rather than being part of the GPU tests,
94+
# because it takes around 30 minutes to run.
95+
parallelism: 16
96+
timeout_in_minutes: 60
97+
retry:
98+
<<: *retry_settings
99+
commands:
100+
- make sudo TARGETS=//tools/gpu:main ARGS="install --latest" || cat /var/log/nvidia-installer.log
101+
- make cuda-12-8-tests ARGS="--cuda_verify_compatibility=true"
102+
agents:
103+
queue: gpu
91104
- <<: *common
92105
label: ":screwdriver: All GPU Drivers Test"
93106
parallelism: 8

test/gpu/cuda_12_8_test.go

Lines changed: 8 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -24,46 +24,22 @@ import (
2424
"gvisor.dev/gvisor/test/gpu/cuda"
2525
)
2626

27-
var testSuiteCompatibility = map[string]cuda.Compatibility{
28-
"0_Introduction": &cuda.NoCrossCompile{},
29-
"1_Utilities": &cuda.NoCrossCompile{},
30-
"2_Concepts_and_Techniques": &cuda.NoCrossCompile{},
31-
"3_CUDA_Features": &cuda.NoCrossCompile{},
32-
"4_CUDA_Libraries": &cuda.NoCrossCompile{},
33-
"5_Domain_Specific": &cuda.NoCrossCompile{},
34-
"6_Performance": &cuda.NoCrossCompile{},
35-
}
27+
var testSuiteCompatibility = map[string]cuda.Compatibility{}
3628

3729
// TODO(b/388095023): Enable these tests once they are tested.
3830
var skippedTestSuites = map[string]string{
39-
"1_Utilities": "TODO: not yet tested",
40-
"2_Concepts_and_Techniques": "TODO: not yet tested",
41-
"3_CUDA_Features": "TODO: not yet tested",
42-
"4_CUDA_Libraries": "TODO: not yet tested",
43-
"5_Domain_Specific": "TODO: not yet tested",
44-
"6_Performance": "TODO: not yet tested",
31+
"3_CUDA_Features": "TODO: not yet tested",
32+
"4_CUDA_Libraries": "TODO: not yet tested",
33+
"5_Domain_Specific": "TODO: not yet tested",
34+
"6_Performance": "TODO: not yet tested",
35+
"7_libNVVM": "TODO: not yet tested",
4536
}
4637

47-
var testCompatibility = map[string]cuda.Compatibility{
48-
"0_Introduction/simpleAttributes": cuda.RequiresFeatures(cuda.FeaturePersistentL2Caching),
49-
"0_Introduction/simpleCUDA2GL": cuda.RequiresFeatures(cuda.FeatureGL),
50-
"0_Introduction/simpleP2P": &cuda.RequiresP2P{},
51-
}
38+
var testCompatibility = map[string]cuda.Compatibility{}
5239

5340
var flakyTests = map[string]struct{}{}
5441

55-
var exclusiveTests = map[string]struct{}{
56-
// Can fail due to
57-
// "launch failed because launch would exceed cudaLimitDevRuntimePendingLaunchCount"
58-
// when running in parallel with other tests.
59-
"3_CUDA_Features/cdpAdvancedQuicksort": {},
60-
61-
// Performance-intensive tests that tend to make other concurrent tests
62-
// flake due to their high resource usage.
63-
"6_Performance/alignedTypes": {},
64-
"6_Performance/transpose": {},
65-
"6_Performance/UnifiedMemoryPerf": {},
66-
}
42+
var exclusiveTests = map[string]struct{}{}
6743

6844
// alwaysSkippedTests don't run at all, ever, and are not verified when
6945
// --cuda_verify_compatibility is set.

0 commit comments

Comments
 (0)