netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Kubernetes benchmarks: Fix wrong path to some GPU container images.

Browse Source 
PiperOrigin-RevId: 704851676
This commit is contained in:
Etienne Perot
2024-12-10 14:45:22 -08:00
committed by gVisor bot
parent d1b164b8e5
commit d29d9acb1f
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/kubernetes/benchmarks/ollama.go
+2 -2
View File
@@ -809,8 +809,8 @@ const (
ollamaPort = 11434
ollamaPodName = "ollama-server"
ollamaServiceName = "ollama-service"
ollamaBenchImage = k8s.ImageRepoPrefix + "benchmarks/gpu/ollama/bench:latest"
ollamaBenchClientImage = k8s.ImageRepoPrefix + "benchmarks/gpu/ollama/client:latest"
ollamaBenchImage = k8s.ImageRepoPrefix + "gpu/ollama/bench:latest"
ollamaBenchClientImage = k8s.ImageRepoPrefix + "gpu/ollama/client:latest"
)
// newOllamaServerPod returns the pod spec for an ollama server.
test/kubernetes/benchmarks/pytorch.go
+2 -2
View File
@@ -50,7 +50,7 @@ const (
train = pytorchTestType("train")
eval = pytorchTestType("eval")
pytorchImage = k8s.ImageRepoPrefix + "benchmarks/pytorch_x86_64:f6f280aeb1b07989"
pytorchImage = k8s.ImageRepoPrefix + "gpu/pytorch_x86_64:latest"
)
type pytorchMode string
@@ -235,7 +235,7 @@ func (p pytorchTest) toPod(namespace *testcluster.Namespace, image string) (*v13
func (p pytorchTest) command() []string {
return []string{
"python",
"python3",
"run.py",
p.module,
"--device", "cuda",