wehub-resource-sync
45 lines
1.5 KiB
C++
45 lines
1.5 KiB
C++
// Copyright (c) 2023 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.
|
|
|
|
#include "paddle/cinn/ir/dim.h"
|
|
#include "paddle/cinn/common/dim_expr_converter.h"
|
|
#include "paddle/cinn/ir/ir.h"
|
|
|
|
namespace cinn {
|
|
namespace ir {
|
|
|
|
const _Dim_* Dim::operator->() const { return As<_Dim_>(); }
|
|
_Dim_* Dim::operator->() { return As<_Dim_>(); }
|
|
|
|
bool _Dim_::IsUniSymbolic() const { return sym_dim.isa<std::string>(); }
|
|
|
|
std::string _Dim_::ToString() const { return symbol::ToString(sym_dim); }
|
|
|
|
Expr _Dim_::GetDimExpr() const { return dim_expr; }
|
|
|
|
Dim _Dim_::Make(const std::string& name, const symbol::DimExpr& sym_dim) {
|
|
auto* n = make_shared<_Dim_>();
|
|
n->name = name;
|
|
n->sym_dim = sym_dim;
|
|
n->dim_expr = common::DimExprConverter().ConvertToIrExpr(sym_dim);
|
|
n->set_type(n->dim_expr.type());
|
|
return Dim(n);
|
|
}
|
|
|
|
Dim::Dim(const std::string& name, const symbol::DimExpr& sym_dim)
|
|
: IrNodeRef(_Dim_::Make(name, sym_dim).self()) {}
|
|
|
|
} // namespace ir
|
|
} // namespace cinn
|