diff --git a/requirements.txt b/requirements.txt index a05cbee..f6438fe 100644 --- 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 index af98820..92460a3 100644 --- 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 index 328d3fa..4fc8626 100644 --- 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(