Skip to content

Commit d2b3758

Browse files
authored
Remove benchmark from tools (#2173)
The benchmark is maintained separately in Azure Devops and the one here is not used. So remove.
1 parent 078b27f commit d2b3758

File tree

8 files changed

+0
-1578
lines changed

8 files changed

+0
-1578
lines changed

onnxscript/_internal/version_utils.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -43,26 +43,6 @@ def transformers_older_than(version: str) -> bool | None:
4343
)
4444

4545

46-
def is_onnxruntime_training() -> bool:
47-
"""Returns True if the onnxruntime is onnxruntime-training."""
48-
try:
49-
from onnxruntime import training # pylint: disable=import-outside-toplevel
50-
51-
assert training
52-
except ImportError:
53-
# onnxruntime not training
54-
return False
55-
56-
try:
57-
from onnxruntime.capi.onnxruntime_pybind11_state import ( # pylint: disable=import-outside-toplevel
58-
OrtValueVector,
59-
)
60-
except ImportError:
61-
return False
62-
63-
return hasattr(OrtValueVector, "push_back_batch")
64-
65-
6646
def onnxruntime_older_than(version: str) -> bool:
6747
"""Returns True if the onnxruntime version is older than the given version."""
6848
import onnxruntime # pylint: disable=import-outside-toplevel

onnxscript/tools/benchmark/__init__.py

Lines changed: 0 additions & 23 deletions
This file was deleted.

0 commit comments

Comments
 (0)