Add Hyperpod cluster tests

aws · Aug 28, 2024 · 289d373 · 289d373
1 parent 766e41a
commit 289d373
Show file tree

Hide file tree

Showing 12 changed files with 747 additions and 3 deletions.
diff --git a/generator/test_case_generator.go b/generator/test_case_generator.go
@@ -227,6 +227,10 @@ var testTypeToTestConfig = map[string][]testConfig{
 			testDir: "./test/awsneuron", terraformDir: "terraform/eks/daemon/awsneuron",
 			targets: map[string]map[string]struct{}{"arc": {"amd64": {}}},
 		},
+		{
+			testDir: "./test/hyperpod", terraformDir: "terraform/eks/daemon/hyperpod",
+			targets: map[string]map[string]struct{}{"arc": {"amd64": {}}},
+		},
 	},
 	"eks_deployment": {
 		{testDir: "./test/metric_value_benchmark"},

diff --git a/terraform/eks/daemon/awsneuron/main.tf b/terraform/eks/daemon/awsneuron/main.tf
@@ -47,10 +47,10 @@ resource "aws_eks_node_group" "this" {
     min_size     = 1
   }
 
-  ami_type       = "AL2_x86_64"
+  ami_type       = var.ami_type
   capacity_type  = "ON_DEMAND"
   disk_size      = 20
-  instance_types = ["t3.medium"]
+  instance_types = [var.instance_type]
 
   depends_on = [
     aws_iam_role_policy_attachment.node_AmazonEC2ContainerRegistryReadOnly,

diff --git a/terraform/eks/daemon/awsneuron/variables.tf b/terraform/eks/daemon/awsneuron/variables.tf
@@ -26,3 +26,12 @@ variable "k8s_version" {
   default = "1.28"
 }
 
+variable "ami_type" {
+  type    = string
+  default = "AL2_x86_64"
+}
+
+variable "instance_type" {
+  type    = string
+  default = "t3.medium"
+}