| 123456789101112131415161718192021222324252627282930313233343536373839 |
- # This backend is maintained by ONNX team. To direct issues
- # to the right people, please tag related GitHub issues with `module: onnx`.
- #
- # Maintainers' Github IDs: wschin, xadupre
- # from torch.onnx._internal.onnxruntime import (
- # is_onnxrt_backend_supported,
- # torch_compile_backend,
- # )
- # from .registry import register_backend
- """
- Placeholder for onnxruntime backend for dynamo
- """
- # def has_onnxruntime():
- # # FIXME: update test/dynamo/test_backends.py to call is_onnxrt_backend_supported()
- # return is_onnxrt_backend_supported()
- # if is_onnxrt_backend_supported():
- # register_backend(name="onnxrt", compiler_fn=torch_compile_backend)
- # else:
- # def information_displaying_backend(*args, **kwargs):
- # raise ImportError(
- # "onnxrt is not registered as a backend. "
- # "Please make sure all dependencies such as "
- # "numpy, onnx, onnxscript, and onnxruntime-training are installed. "
- # "Suggested procedure to fix dependency problem:\n"
- # " (1) pip or conda install numpy onnx onnxscript onnxruntime-training.\n"
- # " (2) Open a new python terminal.\n"
- # " (3) Call the API `torch.onnx.is_onnxrt_backend_supported()`:\n"
- # " (4) If it returns `True`, then you can use `onnxrt` backend.\n"
- # " (5) If it returns `False`, please execute the package importing section in "
- # "torch/onnx/_internal/onnxruntime.py under pdb line-by-line to see which import fails."
- # )
- # register_backend(name="onnxrt", compiler_fn=information_displaying_backend)
|
