Created
April 14, 2022 09:21
-
-
Save danieldk/892b1078f6568367e9b4297c34082497 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
diff --git a/thinc/backends/__init__.py b/thinc/backends/__init__.py | |
index ba4daaa8..1b39deb3 100644 | |
--- a/thinc/backends/__init__.py | |
+++ b/thinc/backends/__init__.py | |
@@ -79,6 +79,10 @@ def _import_extra_cpu_backends(): | |
from thinc_apple_ops import AppleOps | |
except ImportError: | |
pass | |
+ try: | |
+ from thinc_mkl_ops import MKLOps | |
+ except ImportError: | |
+ pass | |
try: | |
from thinc_bigendian_ops import BigEndianOps | |
except ImportError: | |
@@ -97,6 +101,7 @@ def get_ops(name: str, **kwargs) -> Ops: | |
_import_extra_cpu_backends() | |
cls = ops_by_name.get("numpy") | |
cls = ops_by_name.get("apple", cls) | |
+ cls = ops_by_name.get("mkl", cls) | |
cls = ops_by_name.get("bigendian", cls) | |
else: | |
cls = ops_by_name.get(name) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment