maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir] Fix declaration of nano time function in benchmark infra 

In d4555698f8, the name of nano precision timer function has changed from `nano_time` to `nanoTime`, but benchmarks were not updated to reflect that. This change addresses the discrepancy.

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D125217
This commit is contained in:
Denys Shabalin 2022-05-09 12:46:36 +02:00
parent 1b07bd9034
commit 89d4904541
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mlir/benchmark/python/common.py
View File

@ -56,14 +56,14 @@ def get_kernel_func_from_module(module: ir.Module) -> func.FuncOp:
def emit_timer_func() -> func.FuncOp: def emit_timer_func() -> func.FuncOp:
"""Returns the declaration of nano_time function. If nano_time function is """Returns the declaration of nanoTime function. If nanoTime function is
used, the `MLIR_RUNNER_UTILS` and `MLIR_C_RUNNER_UTILS` must be included. used, the `MLIR_RUNNER_UTILS` and `MLIR_C_RUNNER_UTILS` must be included.
""" """
i64_type = ir.IntegerType.get_signless(64) i64_type = ir.IntegerType.get_signless(64)
nano_time = func.FuncOp( nanoTime = func.FuncOp(
"nano_time", ([], [i64_type]), visibility="private") "nanoTime", ([], [i64_type]), visibility="private")
nano_time.attributes["llvm.emit_c_interface"] = ir.UnitAttr.get() nanoTime.attributes["llvm.emit_c_interface"] = ir.UnitAttr.get()
return nano_time return nanoTime
def emit_benchmark_wrapped_main_func(func, timer_func): def emit_benchmark_wrapped_main_func(func, timer_func):