Use func.func in IR instead of just func.

The func prefix will become mandatory. PiperOrigin-RevId: 447660854
tensorflow · May 10, 2022 · 6c9787d · 6c9787d
1 parent 7fd1a26
commit 6c9787d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlir_tests/bef_perf/gen_benchmark_mlir_lib.py b/mlir_tests/bef_perf/gen_benchmark_mlir_lib.py
@@ -44,7 +44,7 @@ def generate_benchmark_mlir(func_name, body):
   body = _indent_lines(body, indentation=4)
 
   return """
-func @{func_name}() {{
+func.func @{func_name}() {{
   tfrt_test.benchmark "{func_name}"()
     duration_secs = 10,
     max_count = 1000000,