[Distributed][auto-parallel] Sharding Specification and rule discovery #1033

	name: Tests

	on:
	push:
	branches: [main]

	pull_request:
	workflow_call:

	jobs:
	tests:
	if: github.repository == 'hidet-org/hidet'
	concurrency:
	group: ${{ github.workflow }}-${{ github.head_ref \|\| github.run_id }}
	cancel-in-progress: true
	runs-on: [self-hosted, Linux, X64, gpu]
	container:
	image: nvidia/cuda:11.8.0-devel-ubuntu20.04
	options: --gpus all
	steps:
	- name: Install dependencies via apt
	run: \|
	apt update && DEBIAN_FRONTEND=noninteractive apt install -y ccache git graphviz

	- uses: actions/checkout@v3

	- name: Set up Python
	uses: actions/setup-python@v4
	with:
	python-version: "3.8"

	- name: Setup cmake
	uses: jwlawson/actions-setup-cmake@v1.13
	with:
	cmake-version: '3.19.x'

	- name: Install dependencies via pip
	run: \|
	python -m pip install --upgrade pip
	pip install torch torchvision torchaudio
	pip install -r requirements.txt
	pip install -r requirements-dev.txt

	- name: Build hidet
	run: \|
	bash scripts/wheel/build_wheel.sh
	WHEEL=$(find ./scripts/wheel/built_wheel -maxdepth 1 -name '*.whl')
	echo "WHEEL_NAME=$WHEEL" >> $GITHUB_ENV
	echo "Built wheel: ${{ env.WHEEL_NAME }}"

	- name: Install hidet
	run: \|
	pip install --no-deps --force-reinstall ${{ env.WHEEL_NAME }}

	- name: List installed packages
	run: \|
	pip list

	# Run tests

	- name: Run tests
	run: \|
	python -m pytest -v --durations=20 --clear-cache ./tests

	# Build the docs

	- name: Install docs dependencies
	run: \|
	pip install -r docs/requirements.txt

	- name: Build docs
	run: \|
	cd docs; make clean; make html

Provide feedback