From dedd5f4a123193da52455c3e0e2aac022a999b87 Mon Sep 17 00:00:00 2001 From: Tian Jin Date: Tue, 9 Jun 2020 02:03:35 +0800 Subject: [PATCH] Minor bugfix. (#166) --- src/Runtime/CMakeLists.txt | 2 +- utils/debug.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/Runtime/CMakeLists.txt b/src/Runtime/CMakeLists.txt index a5f2a55..0e7742c 100644 --- a/src/Runtime/CMakeLists.txt +++ b/src/Runtime/CMakeLists.txt @@ -35,5 +35,5 @@ target_include_directories(PyRuntime PRIVATE ${ONNX_MLIR_SRC_ROOT}) add_dependencies(PyRuntime cruntime) -install(FILES Runtime/DynMemRef.h DESTINATION include) +install(FILES DynMemRef.h DESTINATION include) install(TARGETS cruntime DESTINATION lib) diff --git a/utils/debug.py b/utils/debug.py index d8d987d..d966ac9 100644 --- a/utils/debug.py +++ b/utils/debug.py @@ -22,15 +22,15 @@ if (not os.environ.get('ONNX_MLIR_HOME', None)): VERBOSE = os.environ.get('VERBOSE', False) ONNX_MLIR = os.path.join(os.environ['ONNX_MLIR_HOME'], "bin/onnx-mlir") -# Include runtime directory in python paths, so pyruntime can be imported. +# Include runtime directory in python paths, so PyRuntime can be imported. RUNTIME_DIR = os.path.join(os.environ['ONNX_MLIR_HOME'], "lib") sys.path.append(RUNTIME_DIR) try: - from pyruntime import ExecutionSession + from PyRuntime import ExecutionSession except ImportError: raise ImportError( - "Looks like you did not build the pyruntime target, build it by running `make pyruntime`." + "Looks like you did not build the PyRuntime target, build it by running `make PyRuntime`." )