placement group, availability zone and p3dn.24xlarge support

Author: ajayvohra2005

cfn-template/deeplearning.template

@@ -32,6 +32,11 @@
       "AllowedPattern": "(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})/(\\d{1,2})",
       "Type" : "String"
     },
+    "MyAvailabilityZone" : {
+      "Description" : "My availability zone, e.g. us-east-1d",
+      "AllowedPattern": "[a-z]+-[a-z]+-[1-9]{1}[a-z]{1}",
+      "Type" : "String"
+    },
     "PublicSubnetCIDR" : {
       "Default": "192.168.0.0/27",
       "Description" : "Public Subnet CIDR in MyVpcCIDR",
@@ -69,14 +74,15 @@
     "InstanceType" : {
       "Description" : "The EC2 instance type for workers. For latest GPUs choose p3.xx",
       "Type" : "String",
-      "Default" : "p3.16xlarge",
+      "Default" : "p3dn.24xlarge",
       "AllowedValues" : [
         "p2.16xlarge",
         "p2.8xlarge",
         "p2.xlarge",
         "p3.2xlarge",
         "p3.8xlarge",
         "p3.16xlarge",
+        "p3dn.24xlarge",
         "g2.8xlarge",
         "g2.2xlarge",
         "t2.small",
@@ -236,6 +242,12 @@
     }
   },
   "Resources" : {
+    "ResourcePlacementGroup": {
+  	"Type" : "AWS::EC2::PlacementGroup",
+  	"Properties" : {
+    		"Strategy" : "cluster"
+  	}
+    },
     "ResourceMetadataLambdaFunction": {
       "Type": "AWS::Lambda::Function",
       "DependsOn" : ["MasterQueue"],
@@ -851,6 +863,9 @@
         "DesiredCapacity" : "1",
         "MinSize" : "1",
         "MaxSize" : "1",
+        "PlacementGroup": {
+		"Ref": "ResourcePlacementGroup"
+        },
         "LaunchConfigurationName" : { "Ref" : "MasterLaunchConfig"},
         "VPCZoneIdentifier" : [{ "Ref" : "PublicSubnet"}],
         "NotificationConfiguration" : {
@@ -884,6 +899,9 @@
       "Properties" : {
         "MinSize" : "0",
         "MaxSize" : { "Ref" : "WorkerCount" },
+        "PlacementGroup": {
+		"Ref": "ResourcePlacementGroup"
+        },
         "DesiredCapacity" : { "Ref" : "WorkerCount" },
         "LaunchConfigurationName" : {
           "Ref" : "WorkerLaunchConfig"
@@ -1002,6 +1020,7 @@
       "Properties" : {
           "VpcId" : { "Ref" : "Vpc" },
           "CidrBlock" : { "Ref" : "PrivateSubnetCIDR"},
+          "AvailabilityZone": { "Ref": "MyAvailabilityZone"},
           "Tags" : [
             { "Key" : "Network", "Value" : "Private" },
             { "Key" : "Name", "Value" : { "Ref" : "AWS::StackName" }}