提交 d26a64fc authored 作者: Michael Osthege's avatar Michael Osthege 提交者: Thomas Wiecki

Inherit BaseBLAS Ops from COp

Fixes a performance digression caused by Blas Ops that did not leverage C-thunking.
上级 8b44e8ca
# import theano.tensor.basic as tt # import theano.tensor.basic as tt
from theano import config from theano import config
from theano.gof.op import COp
from theano.gof.params_type import ParamsType from theano.gof.params_type import ParamsType
from theano.scalar import bool as bool_t from theano.scalar import bool as bool_t
...@@ -23,7 +24,7 @@ from theano.tensor.blas import ( ...@@ -23,7 +24,7 @@ from theano.tensor.blas import (
from theano.tensor.opt import in2out from theano.tensor.opt import in2out
class BaseBLAS: class BaseBLAS(COp):
def c_libraries(self): def c_libraries(self):
return ldflags() return ldflags()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论