| # __KERNEL_NAME__ |
| |
| A custom kernel for PyTorch. |
| |
| ## Installation |
| |
| ```bash |
| pip install __REPO_ID__ |
| ``` |
| |
| ## Usage |
| |
| ```python |
| import torch |
| from __KERNEL_NAME_NORMALIZED__ import __KERNEL_NAME_NORMALIZED__ |
|
|
| # Create input tensor |
| x = torch.randn(1024, 1024, device="cuda") |
|
|
| # Run kernel |
| result = __KERNEL_NAME_NORMALIZED__(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 |
| |
