update bc_installer and buildspec

Bayer-Group · Jun 14, 2024 · 780a63e · 780a63e
1 parent e6bd57e
commit 780a63e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/buildspec.yml b/buildspec.yml
@@ -37,6 +37,7 @@ phases:
         AmiId=${AMI_ID:-auto}
         CompilerMacroName=${COMPILER_MACRO_NAME}
         ExistingBatchSubscriptionFilter="${BATCH_FILTER:-none}"
+        GpuAmiId=${GPU_AMI_ID:-auto}
         InstallationName=${INSTALLATION_NAME}
         LauncherBucketName=${LAUNCHER_BUCKET_NAME}
         LogRetentionDays=${LOG_RETENTION_DAYS}

diff --git a/cloudformation/bc_installer.yaml b/cloudformation/bc_installer.yaml
@@ -28,6 +28,7 @@ Metadata:
           default: Batch
         Parameters:
           - AmiId
+          - GpuAmiId
           - RootVolumeSize
           - ScratchVolumeSize
           - MinvCpus
@@ -68,6 +69,14 @@ Parameters:
     Description: GitHub repo containing the BayerCLAW core code.
     Default: "Bayer-Group/BayerCLAW"
 
+  GpuAmiId:
+    Type: String
+    Description: >
+      The ID of a GPU-enabled Amazon Machine Image (AMI) to create EC2 instances with. Enter "auto" to have Batch
+      select an AMI. Otherwise, it is highly recommended that you choose a GPU-enabled, ECS-optimized Amazon Linux 2
+      image to use.
+    Default: "auto"
+
   InstallationName:
     Type: String
     Description: >
@@ -221,6 +230,9 @@ Resources:
           - Name: COMPILER_MACRO_NAME
             Type: PLAINTEXT
             Value: !Ref CompilerMacroName
+          - Name: GPU_AMI_ID
+            Type: PLAINTEXT
+            Value: !Ref GpuAmiId
           - Name: INSTALLATION_NAME
             Type: PLAINTEXT
             Value: !Ref InstallationName