File size: 855 Bytes
8da54a5
 
 
c148255
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
> [!CAUTION]
> Starting from September 13, 2026, we will be removing the "model" type repositories of kernels (e.g., kernels-community/flash-attn3). Make sure you're using a latest version of kernels. If you face any disruption, please report them here: https://github.com/huggingface/kernels/issues/new.

# __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