Views
No views yet
pip install __REPO_ID__1import torch
2from __KERNEL_NAME_NORMALIZED__ import __KERNEL_NAME_NORMALIZED__
3
4# Create input tensor
5x = torch.randn(1024, 1024, device="cuda")
6
7# Run kernel
8result = __KERNEL_NAME_NORMALIZED__(x)1nix develop
2nix run .#build-and-copy1nix develop .#test
2pytest tests/kernels useruv run example.py