| # first-kernel | |
| A custom kernel for PyTorch. | |
| ## Installation | |
| ```bash | |
| pip install drbh/first-kernel | |
| ``` | |
| ## Usage | |
| ```python | |
| import torch | |
| from first_kernel import first_kernel | |
| # Create input tensor | |
| x = torch.randn(1024, 1024, device="cuda") | |
| # Run kernel | |
| result = first_kernel(x) | |
| ``` | |
| ## Development | |
| ### Building | |
| ```bash | |
| nix develop | |
| nix run .#build-and-copy | |
| ``` | |
| ### Testing | |
| ```bash | |
| nix develop .#test | |
| pytest tests/ | |
| ``` | |
| ### Test as a `kernels` user | |
| ```bash | |
| uv run example.py | |
| ``` | |
| ## License | |
| Apache 2.0 | |