hard coding for mammouth in unused code

上级 a0a9ab02
...@@ -89,8 +89,8 @@ _gemm_code = { 'f': _gemm_code_template % { 'gemm':'cblas_sgemm', 'dtype':'float ...@@ -89,8 +89,8 @@ _gemm_code = { 'f': _gemm_code_template % { 'gemm':'cblas_sgemm', 'dtype':'float
def _gemm_rank2(a, x, y, b, z): def _gemm_rank2(a, x, y, b, z):
weave.inline(_gemm_code[z.dtype.char], weave.inline(_gemm_code[z.dtype.char],
['a', 'x', 'y', 'b', 'z'], ['a', 'x', 'y', 'b', 'z'],
headers=['<gsl/gsl_cblas.h>'], headers=['"/home/bergstra/cvs/lgcm/omega/cblas.h"'],
libraries=['cblas','goto', 'g2c']) libraries=['mkl', 'm'])
def _gemm(a, x, y, b, z): def _gemm(a, x, y, b, z):
if len(x.shape) == 2 and len(y.shape) == 2: if len(x.shape) == 2 and len(y.shape) == 2:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论