Skip to content

Add a doc about local development #69

Add a doc about local development

Add a doc about local development #69

Workflow file for this run

name: "Build and test kernel"
on:
push:
branches: [main]
pull_request:
branches: [main]
types: [opened, synchronize, reopened] # trigger on PRs
workflow_dispatch:
jobs:
build:
name: Build kernel
runs-on:
group: aws-g6-12xlarge-plus
steps:
- uses: actions/checkout@v4
- uses: cachix/install-nix-action@v27
with:
nix_path: nixpkgs=channel:nixos-unstable
- uses: cachix/cachix-action@v14
with:
name: kernel-builder
authToken: "${{ secrets.CACHIX_AUTH_TOKEN }}"
env:
USER: github_runner
- name: Build activation kernel
run: ( cd examples/activation && nix build .\#redistributable.torch25-cxx98-cu121-x86_64-linux )
- name: Copy activation kernel
run: cp -rL examples/activation/result activation-kernel
- name: Build cutlass GEMM kernel
run: ( cd examples/cutlass-gemm && nix build .\#redistributable.torch25-cxx98-cu121-x86_64-linux )
- name: Copy cutlass GEMM kernel
run: cp -rL examples/cutlass-gemm/result cutlass-gemm-kernel
- name: Build relu kernel
run: ( cd examples/relu && nix build .\#redistributable.torch25-cxx98-cu121-x86_64-linux )
- name: Copy relu kernel
run: cp -rL examples/relu/result relu-kernel
- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v3
- name: Build Docker image
uses: docker/build-push-action@v5
with:
context: .
file: tests/Dockerfile.test-kernel
platforms: linux/amd64
load: true
push: false
tags: kernel-builder:latest
- name: Run Tests
run: |
docker run --gpus all kernel-builder:latest