[WIP]Add GPU SKUs; clean up (Azure#316)

* Add GPU SKUs; clean upJ * Fix go mod * Update go.mod
mainred · Sep 11, 2020 · d6b3313 · d6b3313
1 parent b9e22be
commit d6b3313
Show file tree

Hide file tree

Showing 7 changed files with 459 additions and 509 deletions.
diff --git a/go.sum b/go.sum
diff --git a/hack/tools/go.mod b/hack/tools/go.mod
@@ -3,6 +3,7 @@ module github.com/Azure/aks-engine/hack/tools
 go 1.12
 
 require (
+	github.com/devigned/pub v0.2.3 // indirect
 	github.com/go-bindata/go-bindata v3.1.2+incompatible
 	github.com/mitchellh/gox v1.0.1
 	github.com/onsi/ginkgo v1.10.1

diff --git a/hack/tools/go.sum b/hack/tools/go.sum
diff --git a/pkg/agent/baker.go b/pkg/agent/baker.go
@@ -321,7 +321,7 @@ func getContainerServiceFuncMap(config *NodeBootstrappingConfiguration) template
 			return cs.Properties.AnyAgentIsLinux()
 		},
 		"IsNSeriesSKU": func(profile *datamodel.AgentPoolProfile) bool {
-			return IsNvidiaEnabledSKU(profile.VMSize)
+			return datamodel.IsNvidiaEnabledSKU(profile.VMSize)
 		},
 		"HasAvailabilityZones": func(profile *datamodel.AgentPoolProfile) bool {
 			return profile.HasAvailabilityZones()