Skip to content

Instantly share code, notes, and snippets.

@danieldk
Created April 14, 2022 09:21
Show Gist options
  • Save danieldk/892b1078f6568367e9b4297c34082497 to your computer and use it in GitHub Desktop.
Save danieldk/892b1078f6568367e9b4297c34082497 to your computer and use it in GitHub Desktop.
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