• Ricardo Vieira's avatar
    Cleanup Fusion rewrites · 2c84b496
    Ricardo Vieira 提交于
    * Move local_add_mul_fusion to `rewriting/elemwise` and remove unused/duplicated TestAddMulFusion tests
    * Use EquilibriumGraphRewriter for local_add_mul_fusion
    * Do not register optional rewrites if tensor__local_elemwise_fusion flag is disabled
    2c84b496
test_math.py 147.5 KB