This repository has been archived by the owner on Aug 7, 2024. It is now read-only.

Add gpu test config to ci/cd #1

	name: 4 GPU Unit Test

	on:
	push:
	branches: [main]
	pull_request:

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

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

	jobs:
	unit_tests_4gpu:
	runs-on: linux.g5.12xlarge.nvidia.gpu
	strategy:
	matrix:
	python-version: ["3.10"]
	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: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -e .
	pip install -e .'[dev]'
	pip install -e .'[test]'
	- name: GPU Tests
	run: NGPU=4 ./test/test_everything.sh

Provide feedback