add nvidia gpu integ test (#399)

aws · Apr 8, 2024 · c638bfa · c638bfa
1 parent 32b91e6
commit c638bfa
Show file tree

Hide file tree

Showing 20 changed files with 1,577 additions and 219 deletions.
diff --git a/environment/metadata.go b/environment/metadata.go
@@ -43,6 +43,7 @@ type MetaData struct {
 	InstanceId                string
 	InstancePlatform          string
 	AgentStartCommand         string
+	EksGpuType                string
 }
 
 type MetaDataStrings struct {
@@ -65,6 +66,7 @@ type MetaDataStrings struct {
 	InstanceId                string
 	InstancePlatform          string
 	AgentStartCommand         string
+	EksGpuType                string
 }
 
 func registerComputeType(dataString *MetaDataStrings) {
@@ -94,6 +96,7 @@ func registerECSData(dataString *MetaDataStrings) {
 func registerEKSData(d *MetaDataStrings) {
 	flag.StringVar(&(d.EKSClusterName), "eksClusterName", "", "EKS cluster name")
 	flag.StringVar(&(d.EksDeploymentStrategy), "eksDeploymentStrategy", "", "Daemon/Replica/Sidecar")
+	flag.StringVar(&(d.EksGpuType), "eksGpuType", "", "nvidia/inferentia")
 }
 
 func registerPluginTestsToExecute(dataString *MetaDataStrings) {
@@ -210,6 +213,7 @@ func fillEKSData(e *MetaData, data *MetaDataStrings) {
 	}
 
 	e.EKSClusterName = data.EKSClusterName
+	e.EksGpuType = data.EksGpuType
 }
 func RegisterEnvironmentMetaDataFlags() *MetaDataStrings {
 	registerComputeType(registeredMetaDataStrings)
@@ -250,6 +254,7 @@ func GetEnvironmentMetaData() *MetaData {
 	metaDataStorage.InstanceId = registeredMetaDataStrings.InstanceId
 	metaDataStorage.InstancePlatform = registeredMetaDataStrings.InstancePlatform
 	metaDataStorage.AgentStartCommand = registeredMetaDataStrings.AgentStartCommand
+	metaDataStorage.EksGpuType = registeredMetaDataStrings.EksGpuType
 
 	return metaDataStorage
 }
diff --git a/generator/test_case_generator.go b/generator/test_case_generator.go
@@ -214,6 +214,10 @@ var testTypeToTestConfig = map[string][]testConfig{
 			targets: map[string]map[string]struct{}{"arc": {"amd64": {}}},
 		},
 		{testDir: "./test/fluent", terraformDir: "terraform/eks/daemon/fluent/windows/2022"},
+		{
+			testDir: "./test/gpu", terraformDir: "terraform/eks/daemon/gpu",
+			targets: map[string]map[string]struct{}{"arc": {"amd64": {}}},
+		},
 	},
 	"eks_deployment": {
 		{testDir: "./test/metric_value_benchmark"},