mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-04 23:59:56 +00:00
dc486d146b
* Make ORT as Pytorch JIT backend LORT likely doesn't work with aten fallback so we only test LORT in its own CI. * Revert changes to enable external CUDA allocator. Will add it later. Revert "Revert changes to enable external CUDA allocator. Will add it later." This reverts commit d5487f2e193014c805505afae8fb577c53667658. Fix external allocator * Relax tolerance and remove commented code * Print more information in CI * Fix pointer * Address comments. 1. Reuse ORT-eager mode's environment. 2. Remove unused ctor. * Use Pytorch master branch as all PRs are merged Fix * Refine based on cpplint feedbacks * Revert changes to allow custom CUDA allocator in public APIs * Use torch.testing.assert_close * Use unittest framework * Switch docker repo * Rename *.cpp to *.cc * Address comments * Add comment * Use same pipeline file for eager and lort pipelines * Address comments * Add yaml comment * Fix cmake files * Address comments * Rename flags, remove printing code, remove dead comment
34 lines
1.6 KiB
Python
34 lines
1.6 KiB
Python
# -------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
# --------------------------------------------------------------------------
|
|
"""
|
|
Ensure that dependencies are available and then load the extension module.
|
|
"""
|
|
import os
|
|
import platform
|
|
import sys
|
|
import warnings
|
|
|
|
from . import _ld_preload # noqa: F401
|
|
|
|
if platform.system() == "Windows":
|
|
from . import version_info
|
|
|
|
# If on Windows, check if this import error is caused by the user not installing the 2019 VC Runtime
|
|
# The VC Redist installer usually puts the VC Runtime dlls in the System32 folder, but it may also be found
|
|
# in some other locations.
|
|
# TODO, we may want to try to load the VC Runtime dlls instead of checking if the hardcoded file path
|
|
# is valid, and raise ImportError if the load fails
|
|
if version_info.vs2019 and platform.architecture()[0] == "64bit":
|
|
system_root = os.getenv("SystemRoot") or "C:\\Windows"
|
|
if not os.path.isfile(os.path.join(system_root, "System32", "vcruntime140_1.dll")):
|
|
warnings.warn("Please install the 2019 Visual C++ runtime and then try again. "
|
|
"If you've installed the runtime in a non-standard location "
|
|
"(other than %SystemRoot%\System32), "
|
|
"make sure it can be found by setting the correct path.")
|
|
|
|
@ONNXRUNTIME_IMPORT_PYTORCH_TO_RESOLVE_DLLS@
|
|
@ONNXRUNTIME_SETDLOPENFLAGS_GLOBAL@
|
|
from .onnxruntime_pybind11_state import * # noqa
|
|
@ONNXRUNTIME_SETDLOPENFLAGS_LOCAL@
|