6 Commits
Author SHA1 Message Date
Etienne PerotandgVisor bot 8516598640 Re-add PyTorch benchmarks into PyTorch image. Update CUDA version.
The PyTorch benchmarks were removed in
https://github.com/google/gvisor/commit/9304ed401fd9604bc36c0436a1132a8b4e3f0851#diff-fd8d6db82d75e1038ed6136c9930c17d6985ff5d22f2ed9e5e8910661de14228
but the Kubernetes PyTorch Kubernetes benchmarks actually depended on them.

PiperOrigin-RevId: 706569220
2024-12-15 21:36:27 -08:00
Etienne PerotandgVisor bot e13cf36ad7 Update all GPU tests to use the ioctl sniffer.
Fixes issue #10885.

PiperOrigin-RevId: 688728104
2024-10-22 16:21:47 -07:00
Andrei VaginandgVisor bot b27d4b5834 images: don't build unused images
Runtime and gpu tests are running on x86_64 only.

PiperOrigin-RevId: 613771002
2024-03-07 18:40:14 -08:00
Etienne PerotandgVisor bot e6a42ae594 Add some PyTorch CUDA tests.
This also adds a repro case for issue #9827, although it is commented out
for now since it doesn't work yet.

PiperOrigin-RevId: 595238934
2024-01-02 16:42:37 -08:00
Ayush RanjanandgVisor bot 9304ed401f Remove unused images.
PiperOrigin-RevId: 574962529
2023-10-19 12:17:56 -07:00
Zach KoopmansandgVisor bot 8a7617aed2 Move pytorch benchmark to GPU directory.
The pytorch image is huge, about 7-8 GB. This doesn't
need to be downloaded with the other benchmark images. Move
it to a GPU directory to speed up benchmark runs.

PiperOrigin-RevId: 566449296
2023-09-18 16:46:52 -07:00