提交 c56c05d2 authored 作者: Frederic's avatar Frederic

Add the opt name as a tag.

上级 34f23400
...@@ -1847,17 +1847,15 @@ blas_optdb.register('local_gemm_to_gemv', ...@@ -1847,17 +1847,15 @@ blas_optdb.register('local_gemm_to_gemv',
15, 'fast_run') 15, 'fast_run')
# After destroyhandler is in but before we try to make elemwise things inplace # After destroyhandler(49.5) but before we try to make elemwise things
# Try to make gemm inplace # inplace (75)
# Also, need to make the gemm optimisation(step 70) happen before the
# fusion of elemwise(step 71)
blas_opt_inplace = in2out(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") 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', 'blas_opt_inplace')
class Dot22Scalar(GemmRelated): class Dot22Scalar(GemmRelated):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论