CUDA kernel registration supports #828
This file contains hidden or 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: Smoke Test | |
| on: | |
| push: | |
| branches: | |
| - main | |
| pull_request: | |
| branches: | |
| - main | |
| jobs: | |
| smoke-test: | |
| runs-on: 4-core-ubuntu-gpu-t4 | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Install CUDA Toolkit | |
| uses: Jimver/cuda-toolkit@v0.2.18 | |
| with: | |
| cuda: '12.4.0' | |
| - name: Install uv | |
| uses: astral-sh/setup-uv@v3 | |
| - name: Set up Python | |
| run: uv python install 3.13 | |
| # - name: Find CUDA version | |
| # run: | | |
| # ls /usr/local | grep cuda | |
| - name: Install package and dependencies | |
| run: uv sync --dev | |
| - name: Clone FACTO source | |
| run: git clone https://github.com/pytorch-labs/FACTO.git | |
| - name: Build and install FACTO | |
| run: cd FACTO && uv pip install . | |
| - name: Run smoke test | |
| run: uv run python -m BackendBench.scripts.main --suite smoke --backend aten | |
| - name: Run FACTO test | |
| run: uv run python -m BackendBench.scripts.main --suite facto --backend aten --ops "add.Tensor" | |
| - name: Run pytest tests | |
| run: uv run pytest test/ |