search

neuralmagic / deepsparse

Sparsity-aware deep learning inference runtime for CPUs
https://neuralmagic.com/deepsparse/
Other
2.94k stars 169 forks source link

[Bug] Fix import bug with numpy_log_softmax #1563

Closed rahul-tuli closed 5 months ago

rahul-tuli commented 5 months ago

This PR fixes a bug with the recently added numpy_log_softmax function, which was causing import errors

Before this PR:

In [1]: from deepsparse.utils import numpy_log_softmax                                                                            
---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-1-056db553cbcd> in <module>
----> 1 from deepsparse.utils import numpy_log_softmax

ImportError: cannot import name 'numpy_log_softmax' from 'deepsparse.utils' (/home/rahul/projects/deepsparse/src/deepsparse/utils/__init__.py)

After this PR:

In [1]: from deepsparse.utils import numpy_log_softmax                                                                               

In [2]: