From 780a63e00558abc85a2387f1449701b94849eff4 Mon Sep 17 00:00:00 2001 From: jetaba Date: Fri, 14 Jun 2024 06:54:34 -0500 Subject: [PATCH] update bc_installer and buildspec --- buildspec.yml | 1 + cloudformation/bc_installer.yaml | 12 ++++++++++++ 2 files changed, 13 insertions(+) diff --git a/buildspec.yml b/buildspec.yml index ca588ba..2973c5e 100644 --- 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 index c0177da..fadd1cc 100644 --- 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