KernelGeneration is a repository created to evaluate GPU kernel metrics, including:
- correctness
- performance
- portability
kernelGen/
├── README.md
├── install.py
└── operators/
├── bf16_layernorm/
└── bf16_matmul/
- install.py — script to install dependencies, copy operators, and clone TritonBench and KernelLLM
- operators/ — contains GPU kernel operators
- TODO/WIP: run_benchmarks.py — script to run all operators and collect results into a JSON file for analysis.
MatMulLayerNorm- WIP add KernelLLM- WIP:
GELU
- TorchInductor
- Triton
- KernelLLM
- WIP: Mako (based on kernelLLM)
Tested on: TritonBench
Run the setup and operator installation script:
python install.py