from kernels.benchmarks import LayerNormBenchmark, RMSNormBenchmark class LayerNorm(LayerNormBenchmark): pass class RMSNorm(RMSNormBenchmark): pass