Kernels
layer-norm / benchmarks /benchmark.py
gueraf's picture
fork of kernels-community/layer-norm @ main + torch29-cxx11-cu128 builds from odysseyml/flash-attention odyssey-v2.8.3-fused-1 (sm80/90/100/120, community signature, fwd-only)
4ba2d64 verified
Raw
History Blame Contribute Delete
160 Bytes
from kernels.benchmarks import LayerNormBenchmark, RMSNormBenchmark
class LayerNorm(LayerNormBenchmark):
pass
class RMSNorm(RMSNormBenchmark):
pass