Skip to content

Commit

Permalink
Merge pull request #211 from containers/arm-test-environment
Browse files Browse the repository at this point in the history
ARM based test environment
  • Loading branch information
rhatdan authored Apr 9, 2024
2 parents 0ca5150 + 4be8423 commit dd4cf14
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 2 deletions.
8 changes: 7 additions & 1 deletion .github/scripts/provision.sh
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,14 @@

set -x

if [[ "$(uname -m)" == "aarch64" ]]; then
arch="arm64"
else
arch="amd64"
fi

dnf install -y podman wget \
https://s3.us-east-2.amazonaws.com/amazon-ssm-us-east-2/latest/linux_amd64/amazon-ssm-agent.rpm
https://s3.us-east-2.amazonaws.com/amazon-ssm-us-east-2/latest/linux_${arch}/amazon-ssm-agent.rpm
dnf clean all

wget -P locallm/models https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGUF/resolve/main/llama-2-7b-chat.Q5_K_S.gguf
Expand Down
17 changes: 16 additions & 1 deletion .github/workflows/testing-framework.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@ on:

env:
TF_VAR_aws_region: "eu-west-2"
TF_VAR_aws_instance_type: "t3.large"
TF_VAR_aws_ami_owners: '["125523088429"]'
TF_VAR_aws_ami_name: '["Fedora-Cloud-Base-*"]'
TF_VAR_aws_volume_size: 100
Expand All @@ -31,6 +30,16 @@ jobs:
integration-tests:
if: github.repository == 'containers/ai-lab-recipes'
runs-on: ubuntu-latest
strategy:
fail-fast: false
matrix:
include:
- arch: amd64
aws_image_type: t3a.medium
aws_ami_architecture: x86_64
- arch: arm64
aws_image_type: m7g.medium
aws_ami_architecture: arm64
steps:
- uses: actions/[email protected]
with:
Expand Down Expand Up @@ -58,6 +67,9 @@ jobs:
id: up
run: terraform apply -auto-approve -lock=false
working-directory: terraform-test-environment-module
env:
TF_VAR_aws_instance_type: ${{ matrix.aws_image_type }}
TF_VAR_aws_ami_architecture: ${{ matrix.aws_ami_architecture }}

- name: Terraform Output
id: terraform-output
Expand Down Expand Up @@ -89,6 +101,9 @@ jobs:
if: always()
run: terraform destroy -auto-approve -lock=false
working-directory: terraform-test-environment-module
env:
TF_VAR_aws_instance_type: ${{ matrix.aws_image_type }}
TF_VAR_aws_ami_architecture: ${{ matrix.aws_ami_architecture }}

- name: Publish Job Results to Slack
id: slack
Expand Down

0 comments on commit dd4cf14

Please sign in to comment.