mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-24 04:06:03 +00:00
Make it possible to override the driver installer daemonset url from test-infra.
This commit is contained in:
parent
f53c26a7fb
commit
a959ae636b
@ -17,6 +17,7 @@ limitations under the License.
|
|||||||
package scheduling
|
package scheduling
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -163,7 +164,12 @@ func testNvidiaGPUsOnCOS(f *framework.Framework) {
|
|||||||
framework.Logf("Cluster is running on COS. Proceeding with test")
|
framework.Logf("Cluster is running on COS. Proceeding with test")
|
||||||
|
|
||||||
if f.BaseName == "device-plugin-gpus" {
|
if f.BaseName == "device-plugin-gpus" {
|
||||||
dsYamlUrl = "https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/daemonset.yaml"
|
dsYamlUrlFromEnv := os.Getenv("NVIDIA_DRIVER_INSTALLER_DAEMONSET")
|
||||||
|
if dsYamlUrlFromEnv != "" {
|
||||||
|
dsYamlUrl = dsYamlUrlFromEnv
|
||||||
|
} else {
|
||||||
|
dsYamlUrl = "https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/daemonset.yaml"
|
||||||
|
}
|
||||||
gpuResourceName = framework.NVIDIAGPUResourceName
|
gpuResourceName = framework.NVIDIAGPUResourceName
|
||||||
podCreationFunc = makeCudaAdditionDevicePluginTestPod
|
podCreationFunc = makeCudaAdditionDevicePluginTestPod
|
||||||
} else {
|
} else {
|
||||||
@ -172,6 +178,7 @@ func testNvidiaGPUsOnCOS(f *framework.Framework) {
|
|||||||
podCreationFunc = makeCudaAdditionTestPod
|
podCreationFunc = makeCudaAdditionTestPod
|
||||||
}
|
}
|
||||||
|
|
||||||
|
framework.Logf("Using %v", dsYamlUrl)
|
||||||
// Creates the DaemonSet that installs Nvidia Drivers.
|
// Creates the DaemonSet that installs Nvidia Drivers.
|
||||||
// The DaemonSet also runs nvidia device plugin for device plugin test.
|
// The DaemonSet also runs nvidia device plugin for device plugin test.
|
||||||
ds, err := framework.DsFromManifest(dsYamlUrl)
|
ds, err := framework.DsFromManifest(dsYamlUrl)
|
||||||
|
Loading…
Reference in New Issue
Block a user