Add device bridge support for HIP and CUDA. #10

	name: CI

	on:
	workflow_dispatch:
	pull_request:
	push:
	branches:
	- main

	concurrency:
	# A PR number if a pull request and otherwise the commit hash. This cancels
	# queued and in-progress runs for the same PR (presubmit) or commit
	# (postsubmit). The workflow name is prepended to avoid conflicts between
	# different workflows.
	group: ${{ github.workflow }}-${{ github.event.number \|\| github.sha }}
	cancel-in-progress: true

	jobs:
	test:
	name: "Unit Tests and Type Checking"
	strategy:
	matrix:
	version: [3.11]
	os: [ubuntu-latest]
	runs-on: ${{matrix.os}}
	env:
	PIP_CACHE_DIR: "${{ github.workspace }}/.pip-cache"
	steps:
	- name: "Setting up Python"
	id: setup_python
	uses: actions/setup-python@v3
	with:
	python-version: ${{matrix.version}}

	- name: "Checkout Code"
	uses: actions/checkout@v3

	- name: Cache Pip Packages
	uses: actions/cache@v4
	id: cache-pip
	with:
	path: ${{ env.PIP_CACHE_DIR }}
	key: pip-${{ steps.setup_python.outputs.python-version }}-${{ hashFiles('*requirements.txt') }}

	- name: Install pip deps
	run: \|
	python -m pip install --no-compile --upgrade pip
	# Note: We install in three steps in order to satisfy requirements
	# from non default locations first. Installing the PyTorch CPU
	# wheels saves multiple minutes and a lot of bandwidth on runner setup.
	pip install --no-compile -r pytorch-cpu-requirements.txt
	pip install -r requirements.txt -e .

	- name: Run unit tests
	if: ${{ !cancelled() }}
	run: \|
	pytest -n 4 .

	- name: MyPy Type Checking
	if: ${{ !cancelled() }}
	run: \|
	mypy

Provide feedback