EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[sgl-kernel] support hadamard (#11663)

Browse Source
This commit is contained in:
Fan Yin
2025-10-16 10:00:44 +08:00
committed by GitHub
parent 868403f642
commit 3289da5b41
7 changed files with 147 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
sgl-kernel/python/sgl_kernel/hadamard.py Normal file
View File

@@ -0,0 +1,21 @@
import torch
def hadamard_transform(x: torch.Tensor, scale: float = 1.0) -> torch.Tensor:
return torch.ops.sgl_kernel.fast_hadamard_transform.default(x, scale)
def hadamard_transform_12n(x: torch.Tensor, scale: float = 1.0) -> torch.Tensor:
return torch.ops.sgl_kernel.fast_hadamard_transform_12N.default(x, scale)
def hadamard_transform_20n(x: torch.Tensor, scale: float = 1.0) -> torch.Tensor:
return torch.ops.sgl_kernel.fast_hadamard_transform_20N.default(x, scale)
def hadamard_transform_28n(x: torch.Tensor, scale: float = 1.0) -> torch.Tensor:
return torch.ops.sgl_kernel.fast_hadamard_transform_28N.default(x, scale)
def hadamard_transform_40n(x: torch.Tensor, scale: float = 1.0) -> torch.Tensor:
return torch.ops.sgl_kernel.fast_hadamard_transform_40N.default(x, scale)