提交 95fa97cc authored 作者: Frederic Bastien's avatar Frederic Bastien

always register local_elemwise_fusion, but not in fast_run by default. Allow the…

always register local_elemwise_fusion, but not in fast_run by default. Allow the test to work and test it event if not enabled.
上级 565c5cca
...@@ -1369,10 +1369,11 @@ class FusionOptimizer(Optimizer): ...@@ -1369,10 +1369,11 @@ class FusionOptimizer(Optimizer):
if config.getboolean('tensor_opt.local_elemwise_fusion'): if config.getboolean('tensor_opt.local_elemwise_fusion'):
_logger.debug("enabling optimization: fusion elemwise") _logger.debug("enabling optimization fusion elemwise in fast_run")
compile.optdb.register('elemwise_fusion', FusionOptimizer(), 71.00, 'fast_run', 'fusion') compile.optdb.register('elemwise_fusion', FusionOptimizer(), 71.00, 'fast_run', 'fusion', 'local_elemwise_fusion')
else: else:
_logger.debug("not enabling optimization: fusion elemwise") _logger.debug("not enabling optimization fusion elemwise in fast_run")
compile.optdb.register('elemwise_fusion', FusionOptimizer(), 71.00, 'fusion', 'local_elemwise_fusion')
# def make_composite(inputs, outputs): # def make_composite(inputs, outputs):
# scalar_inputs = [scalar.Scalar(dtype = i.type.dtype)() for i in inputs] # scalar_inputs = [scalar.Scalar(dtype = i.type.dtype)() for i in inputs]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论