• Yu Cheng's avatar
    [Example] Introduce autotuning example for GEMM with enhanced configuration options (#360) · d4194222
    Yu Cheng authored
    * Added a new example script `example_gemm_autotune.py` to demonstrate autotuning for matrix multiplication (GEMM) using TileLang.
    * Implemented functions for generating configurations, selecting the best configuration, and benchmarking performance.
    * Refactored the existing `matmul` function to support dynamic configuration parameters and improved kernel compilation.
    * Updated the main execution block to include command-line argument parsing for matrix dimensions and autotuning options.
    * Enhanced the example to validate results against a reference implementation, ensuring correctness in matrix multiplication operations.
    d4194222
example_gemm.py 1.36 KB