提交 7dd9f409 authored 作者: Frederic's avatar Frederic

simplify call.

上级 d6651735
...@@ -135,7 +135,7 @@ import numpy.distutils.__config__ ...@@ -135,7 +135,7 @@ import numpy.distutils.__config__
from theano.configparser import config, AddConfigVar, StrParam from theano.configparser import config, AddConfigVar, StrParam
from theano.gof import (utils, Op, view_roots, DestroyHandler, from theano.gof import (utils, Op, view_roots, DestroyHandler,
local_optimizer, Optimizer, LocalOptGroup, local_optimizer, Optimizer,
InconsistencyError, toolbox, SequenceDB, InconsistencyError, toolbox, SequenceDB,
EquilibriumOptimizer, Apply, EquilibriumOptimizer, Apply,
ReplacementDidntRemovedError) ReplacementDidntRemovedError)
...@@ -1777,9 +1777,10 @@ blas_optdb.register('local_gemm_to_gemv', ...@@ -1777,9 +1777,10 @@ blas_optdb.register('local_gemm_to_gemv',
# Try to make gemm inplace # Try to make gemm inplace
# Also, need to make the gemm optimisation(step 70) happen before the # Also, need to make the gemm optimisation(step 70) happen before the
# fusion of elemwise(step 71) # fusion of elemwise(step 71)
blas_opt_inplace = in2out(LocalOptGroup(local_inplace_gemm, blas_opt_inplace = in2out(local_inplace_gemm,
local_inplace_gemv, local_inplace_gemv,
local_inplace_ger)) local_inplace_ger,
name="blas_opt_inplace")
optdb.register('InplaceBlasOpt', optdb.register('InplaceBlasOpt',
blas_opt_inplace, blas_opt_inplace,
70.0, 'fast_run', 'inplace') 70.0, 'fast_run', 'inplace')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论