// Copyright (c) 2021 CINN Authors. All Rights Reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. #pragma once #include "paddle/cinn/ir/ir.h" namespace cinn { namespace optim { /** * Simplify the expression. * The following cases are supported: * a + 0 => a * a*0 => 0 * A[i*0+2*a+3*a+1+2] => A[5*a+3] * * This only works on the simple IR nodes such as Load, Store, and the math * operators such as Add, Sub and so on. */ void Simplify(Expr *expr); void SimplifyForLoops(Expr *expr); void SimplifyBlocks(Expr *expr); } // namespace optim } // namespace cinn