Minor bugfix. (#166)

This commit is contained in:
Tian Jin 2020-06-09 02:03:35 +08:00 committed by GitHub
parent bb17fa965f
commit dedd5f4a12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions

View File

@ -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)

View File

@ -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`."
)