hard coding for mammouth in unused code

上级 a0a9ab02
...@@ -87,10 +87,10 @@ _gemm_code = { 'f': _gemm_code_template % { 'gemm':'cblas_sgemm', 'dtype':'float ...@@ -87,10 +87,10 @@ _gemm_code = { 'f': _gemm_code_template % { 'gemm':'cblas_sgemm', 'dtype':'float
'd': _gemm_code_template % { 'gemm':'cblas_dgemm', 'dtype':'double'}} 'd': _gemm_code_template % { 'gemm':'cblas_dgemm', 'dtype':'double'}}
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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论