Support compute dtype in spmd fsdp #52

	name: Unit Tests

	on:
	workflow_dispatch:
	pull_request:
	push:
	branches:
	- main
	tags:
	- r[0-9]+.[0-9]+


	jobs:
	run-shell-script:
	runs-on: self-hosted

	steps:
	- name: Checkout code
	uses: actions/checkout@v3

	- name: Run unit test
	run: \|
	docker pull $UT_IMAGE
	docker run -v $PWD:$PWD -w $PWD --net host --ipc host --shm-size 80G -t --rm --gpus all $UT_IMAGE bash -c '
	pip install -r requirements/requirements-test.txt && \
	git config --global --add safe.directory $PWD && \
	pip install -e . && \
	bash tests/run_ut.sh'
	env:
	UT_IMAGE: ${{ secrets.UT_IMAGE }}

Provide feedback