Skip to content

Commit

Permalink
.github: Add basic test workflow
Browse files Browse the repository at this point in the history
Signed-off-by: Eli Uriegas <[email protected]>
  • Loading branch information
seemethere committed Apr 10, 2024
1 parent 0a8ebf9 commit 72e9cbe
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions .github/workflows/test.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
name: Run compile tests

on:
pull_request:
push:
branches:
- main

jobs:
test-cuda:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux.g5.4xlarge.nvidia.gpu
gpu-arch-type: cuda
gpu-arch-version: "12.1"
script: |
pip install --extra-index-url https://download.pytorch.org/whl/nightly/cu121 -r requirements.txt
python generate.py --compile --cuda

0 comments on commit 72e9cbe

Please sign in to comment.