diff --git a/mindspore/python/mindspore/_extends/parallel_compile/akg_compiler/akg_process.py b/mindspore/python/mindspore/_extends/parallel_compile/akg_compiler/akg_process.py index c377136d07e9e4bb963524a1f9fdba40284210fd..25344d8b87def8e4e97032c61866f5d3ebc13e2a 100644 --- a/mindspore/python/mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +++ b/mindspore/python/mindspore/_extends/parallel_compile/akg_compiler/akg_process.py @@ -127,8 +127,7 @@ def _compile_akg_mlir_task_default(json_strs, attrs, driver): if not os.path.isfile(info_path): raise FileNotFoundError(f"Can not compile non-existing file \"{info_path}\"") # Compile json str with AKG - bisheng_cpp_path = os.getenv("BISHENG_CPP_PATH", default="") - compiler = driver(input_file=info_path, output_dir=kernel_meta_dir, bisheng_tools_dir=bisheng_cpp_path, + compiler = driver(input_file=info_path, output_dir=kernel_meta_dir, dynamic_shape=_is_input_shape_dynamic(json_desc)) try: compiler.compile()