Kernels:
Trusted publisher
| from kernels.benchmarks import LayerNormBenchmark, RMSNormBenchmark | |
| class LayerNorm(LayerNormBenchmark): | |
| pass | |
| class RMSNorm(RMSNormBenchmark): | |
| pass | |
| from kernels.benchmarks import LayerNormBenchmark, RMSNormBenchmark | |
| class LayerNorm(LayerNormBenchmark): | |
| pass | |
| class RMSNorm(RMSNormBenchmark): | |
| pass | |