提交 bce0954f authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make gemv perform handle float16 (and other non-blas types).

上级 57a52cab
......@@ -417,7 +417,8 @@ class Gemv(Op):
def perform(self, node, inputs, out_storage):
y, alpha, A, x, beta = inputs
if have_fblas and y.shape[0] != 0 and x.shape[0] != 0:
if (have_fblas and y.shape[0] != 0 and x.shape[0] != 0 and
y.dtype in _blas_gemv_fns):
gemv = _blas_gemv_fns[y.dtype]
if (A.shape[0] != y.shape[0] or A.shape[1] != x.shape[0]):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论