From b0308816b2c7929fbccbea690e5d3a200f240201 Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Wed, 4 May 2022 04:28:11 -0400 Subject: [PATCH] [0.12.1 cherrypick] ORT scheduler fix (#347) * Add scheduler property to ort_engine (#345) * Update ort_engine.py --- src/deepsparse/benchmark/ort_engine.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/deepsparse/benchmark/ort_engine.py b/src/deepsparse/benchmark/ort_engine.py index 10da3e36a9..3765aa87b0 100644 --- a/src/deepsparse/benchmark/ort_engine.py +++ b/src/deepsparse/benchmark/ort_engine.py @@ -186,6 +186,13 @@ def num_cores(self) -> int: """ return self._num_cores if self._num_cores else NUM_CORES + @property + def scheduler(self): + """ + :return: The kind of scheduler to execute with + """ + return None + def run( self, inp: List[numpy.ndarray],