make sure benchmark_* imports are the functions, not modules

neuralmagic / deepsparse

Sparsity-aware deep learning inference runtime for CPUs

https://neuralmagic.com/deepsparse/

Other

2.94k stars 169 forks source link

make sure benchmark_* imports are the functions, not modules #1593

Closed bfineran closed 5 months ago

bfineran commented 5 months ago

makes sure that top level benchmark_* imports are the functions, not modules (ie file names)

>>> from deepsparse import benchmark_model, benchmark_pipeline
>>> benchmark_model
<function benchmark_model at 0x7f4b5434faf0>
>>> benchmark_pipeline
<function benchmark_pipeline at 0x7f4b6b368430>