mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-25 07:14:16 +08:00
Lei Wang
e23bce79d6
* Refactor benchmarking calls to remove warmup parameter in do_bench function Updated multiple examples to streamline benchmarking by removing the warmup parameter from the do_bench function calls. This change simplifies the function signature while maintaining the backend specification for performance measurement. * Update regression performance parameters across multiple examples to support larger input sizes Modified the `run_regression_perf` function in various examples to set default parameters for matrix dimensions (M, N, K) and heads to 4096 or 32, enhancing the benchmarking capabilities for larger datasets. This change aims to improve performance testing and ensure consistency across examples.
11 lines
217 B
Python
11 lines
217 B
Python
import tilelang.testing
|
|
import example_dynamic
|
|
|
|
|
|
def regression_example_dynamic():
|
|
tilelang.testing.process_func(example_dynamic.run_regression_perf)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.regression()
|