"""Smoke example for galqiwi/flute_kernels."""

import torch
from kernels import get_kernel

flute = get_kernel("galqiwi/flute_kernels", version=1)

x = torch.randn(4, 4096, device="cuda", dtype=torch.float16)
y = flute.hadamard_transform(x, False)
print(y.shape, y.dtype)