Skip to content

Commit b030881

Browse files
authored
[0.12.1 cherrypick] ORT scheduler fix (#347)
* Add scheduler property to ort_engine (#345) * Update ort_engine.py
1 parent 32addc6 commit b030881

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/deepsparse/benchmark/ort_engine.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -186,6 +186,13 @@ def num_cores(self) -> int:
186186
"""
187187
return self._num_cores if self._num_cores else NUM_CORES
188188

189+
@property
190+
def scheduler(self):
191+
"""
192+
:return: The kind of scheduler to execute with
193+
"""
194+
return None
195+
189196
def run(
190197
self,
191198
inp: List[numpy.ndarray],

0 commit comments

Comments
 (0)