Update for CBlas changes in Thinc 8.1.0.dev2 (#21)

explosion · Jun 16, 2022 · a762bd7 · a762bd7
1 parent 32deb3a
commit a762bd7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 cython>=0.25
 numpy>=1.21.0
-thinc>=8.1.0.dev0,<9.0.0
+thinc>=8.1.0.dev2,<9.0.0
 # dev
 pytest>=5.2.0
diff --git a/setup.cfg b/setup.cfg
@@ -28,7 +28,7 @@ include_package_data = true
 python_requires = >=3.6
 install_requires =
     numpy>=1.21.0
-    thinc>=8.1.0.dev0,<9.0.0
+    thinc>=8.1.0.dev2,<9.0.0
 
 [options.entry_points]
 thinc_ops =

diff --git a/thinc_apple_ops/ops.pyx b/thinc_apple_ops/ops.pyx
@@ -1,6 +1,6 @@
 from typing import Optional
 import numpy
-from thinc.backends.cblas cimport CBlas
+from thinc.backends.cblas cimport CBlas, set_saxpy, set_sgemm
 from thinc.backends.numpy_ops import NumpyOps
 from thinc.types import Floats2d
 from . import blas
@@ -15,8 +15,8 @@ class AppleOps(NumpyOps):
 
     def cblas(self) -> CBlas:
         cdef CBlas cblas = CBlas()
-        cblas.set_saxpy(saxpy)
-        cblas.set_sgemm(sgemm)
+        set_saxpy(cblas, saxpy)
+        set_sgemm(cblas, sgemm)
         return cblas
 
     def gemm(