github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-29 14:37:00 +00:00
Code Issues Projects Releases Wiki Activity

Only create one pod per gpu node in E2E test

Browse Source
This commit is contained in:
Richard Chen 2019-04-26 16:19:17 -07:00
parent 0f9077d0af
commit 9368b2ce87
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
test/e2e/scheduling/nvidia-gpus.go
View File

@ -168,11 +168,23 @@ func SetupNVIDIAGPUNode(f *framework.Framework, setupResourceGatherer bool) *fra
return rsgather
}
func getGPUsPerPod() int64 {
var gpusPerPod int64
gpuPod := makeCudaAdditionDevicePluginTestPod()
for _, container := range gpuPod.Spec.Containers {
if val, ok := container.Resources.Limits[gpuResourceName]; ok {
gpusPerPod += (&val).Value()
}
}
return gpusPerPod
}
func testNvidiaGPUs(f *framework.Framework) {
rsgather := SetupNVIDIAGPUNode(f, true)
e2elog.Logf("Creating as many pods as there are Nvidia GPUs and have the pods run a CUDA app")
gpuPodNum := getGPUsAvailable(f) / getGPUsPerPod()
e2elog.Logf("Creating %d pods and have the pods run a CUDA app", gpuPodNum)
podList := []*v1.Pod{}
for i := int64(0); i < getGPUsAvailable(f); i++ {
for i := int64(0); i < gpuPodNum; i++ {
podList = append(podList, f.PodClient().Create(makeCudaAdditionDevicePluginTestPod()))
}
e2elog.Logf("Wait for all test pods to succeed")