Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Switch to GPU Instances for GSLC Jobs #2332

Draft
wants to merge 15 commits into
base: develop
Choose a base branch
from
Draft
4 changes: 2 additions & 2 deletions .github/workflows/deploy-enterprise-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -88,12 +88,12 @@ jobs:
job_files: >-
job_spec/INSAR_ISCE_BURST.yml
job_spec/SRG_GSLC_CPU.yml
instance_types: r6id.xlarge,r6id.2xlarge,r6id.4xlarge,r6id.8xlarge,r6idn.xlarge,r6idn.2xlarge,r6idn.4xlarge,r6idn.8xlarge
instance_types: g6.2xlarge
jtherrmann marked this conversation as resolved.
Show resolved Hide resolved
default_max_vcpus: 640
expanded_max_vcpus: 640
required_surplus: 0
security_environment: ASF
ami_id: /aws/service/ecs/optimized-ami/amazon-linux-2023/recommended/image_id
ami_id: /aws/service/ecs/optimized-ami/amazon-linux-2/gpu/recommended/image_id
distribution_url: ''

environment:
Expand Down
2 changes: 2 additions & 0 deletions apps/workflow-cf.yml.j2
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,8 @@ Resources:
ResourceRequirements:
- Type: VCPU
Value: "{{ task['vcpu'] }}"
- Type: GPU
Value: 1
- Type: MEMORY
Value: "{{ task['memory'] }}"
Command:
Expand Down
4 changes: 3 additions & 1 deletion job_spec/SRG_GSLC_CPU.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,17 +27,19 @@ SRG_GSLC_CPU:
tasks:
- name: ''
image: ghcr.io/asfhyp3/hyp3-back-projection
image_tag: 0.5.2.gpu
command:
- ++process
- back_projection
- --gpu
- --bucket
- '!Ref Bucket'
- --bucket-prefix
- Ref::bucket_prefix
- Ref::granules
timeout: 10800
vcpu: 1
memory: 31500
memory: 30500
secrets:
- EARTHDATA_USERNAME
- EARTHDATA_PASSWORD