[Operators] Conv2d fp16 implicit gemm kernel #830
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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/[email protected] | |
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 }} | |
# 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 |