add node-accelerators flag

Author: upodroid

kubetest2-gce/deployer/common.go

@@ -146,6 +146,9 @@ func (d *deployer) buildEnv() []string {
 	// target the right network
 	env = append(env, fmt.Sprintf("NETWORK=%s", d.network))
 
+	// NODE_ACCELERATORS is used by kube-up.sh script to set type and number of GPUs
+	env = append(env, fmt.Sprintf("NODE_ACCELERATORS=%s", d.NodeAccelerators))
+
 	if d.EnableCacheMutationDetector {
 		env = append(env, "ENABLE_CACHE_MUTATION_DETECTOR=true")
 	}

kubetest2-gce/deployer/deployer.go

@@ -91,6 +91,7 @@ type deployer struct {
 	CreateCustomNetwork         bool   `desc:"Sets the environment variable CREATE_CUSTOM_NETWORK=true during deployment."`
 	NodeScopes                  string `desc:"Sets the NODE_SCOPES environment variable during deployment."`
 	NodeServiceAccount          string `desc:"Sets the KUBE_GCE_NODE_SERVICE_ACCOUNT environment variable during deployment."`
+	NodeAccelerators            string `desc:"Sets the NODE_ACCELERATORS environment variable during deployment."`
 	CloudProvider               string `desc:"Sets the CLOUD_PROVIDER environment variable during deployment."`
 	FeatureGates                string `desc:"Sets the KUBE_FEATURE_GATES environment variable during deployment."`