update

bytedance · Aug 24, 2024 · ceae8bd · ceae8bd
1 parent 42a4e54
commit ceae8bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/compiler/python/byteir/compile.py b/compiler/python/byteir/compile.py
@@ -1,7 +1,7 @@
 from enum import Enum
 from pathlib import Path
 import os
-from shutil import copymode
+from typing import Union
 
 from . import ir
 from .passmanager import PassManager

diff --git a/tests/numerical_test/execute.py b/tests/numerical_test/execute.py
@@ -149,9 +149,9 @@ def compile_and_run_mlir(mhlo_file, target, workdir, verbose, mode="numerical",
         torch_outputs = data_generator.generate_torch_outputs(cur_device)
 
         if mode == "numerical":
-            brt_backend.run(torch_inputs, torch_outputs)
+            brt_backend.run_with_outputs(torch_inputs, torch_outputs)
         else:
-            avg_time = brt_backend.profile(torch_inputs, torch_outputs)
+            avg_time = brt_backend.profile_with_outputs(torch_inputs, torch_outputs)
             return TestResult(
                 unique_name=unique_name,
                 compilation_error=None,
@@ -246,9 +246,9 @@ def compile_and_run_torch(test, target, workdir, verbose, mode="numerical"):
     try:
         brt_backend = BRTBackend(output_mlir_file_name, cur_device)
         if mode == "numerical":
-            brt_backend.run(torch_inputs, torch_outputs)
+            brt_backend.run_with_outputs(torch_inputs, torch_outputs)
         else:
-            avg_time = brt_backend.profile(torch_inputs, torch_outputs)
+            avg_time = brt_backend.profile_with_outputs(torch_inputs, torch_outputs)
             return TestResult(
                 unique_name=unique_name,
                 compilation_error=None,