Adding quantization support in torchtune #370

	name: Multi-GPU Recipe Tests

	on:
	push:
	branches: [ main ]
	pull_request:
	workflow_dispatch:

	concurrency:
	group: recipe-test-multi-gpu-${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_number \|\| github.ref }}
	cancel-in-progress: true

	permissions:
	id-token: write
	contents: read

	defaults:
	run:
	shell: bash -l -eo pipefail {0}

	jobs:
	recipe_test_multi_gpu:
	runs-on: linux.8xlarge.nvidia.gpu
	strategy:
	matrix:
	python-version: ['3.8', '3.9', '3.10', '3.11']
	steps:
	- name: Check out repo
	uses: actions/checkout@v3
	- name: Setup conda env
	uses: conda-incubator/setup-miniconda@v2
	with:
	auto-update-conda: true
	miniconda-version: "latest"
	activate-environment: test
	python-version: ${{ matrix.python-version }}
	- name: Update pip
	run: python -m pip install --upgrade pip
	- name: configure aws credentials
	id: aws_creds
	uses: aws-actions/configure-aws-credentials@v1.7.0
	with:
	role-to-assume: arn:aws:iam::308535385114:role/gha_workflow_torchtune_pytorch-multimodal
	aws-region: us-east-1
	- name: Install S3 CLI
	run: \|
	python -m pip install awscli==1.32.6
	- name: Install dependencies
	run: \|
	python -m pip install torch
	python -m pip install -e ".[dev]"
	python -m pip install lm-eval==0.4.*
	- name: Run recipe tests with coverage
	run: pytest tests -m integration_test --silence-s3-logs --cov=. --cov-report=xml --durations=20 -vv
	- name: Upload Coverage to Codecov
	uses: codecov/codecov-action@v3

Provide feedback