benchmarks/benchmark.py

from kernels.benchmarks import LayerNormBenchmark, RMSNormBenchmark


class LayerNorm(LayerNormBenchmark):
    pass


class RMSNorm(RMSNormBenchmark):
    pass