wehub-resource-sync
55 lines
2.1 KiB
C++
55 lines
2.1 KiB
C++
// Copyright (c) 2024 PaddlePaddle 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 <optional>
|
|
#include <set>
|
|
#include <string>
|
|
#include <vector>
|
|
#include "paddle/pir/include/core/program.h"
|
|
#include "paddle/pir/include/core/value.h"
|
|
|
|
namespace pir {
|
|
namespace utils {
|
|
namespace name_analysis {
|
|
|
|
std::vector<std::string> GetValueAllNames(Value value);
|
|
std::string GetValueFirstName(Value value);
|
|
std::optional<std::string> TryGetValueFirstName(Value value);
|
|
Value GetParameterValueByName(const Program &program, const std::string &name);
|
|
std::unordered_map<std::string, Value> GetAllParameterValues(
|
|
const Program &program);
|
|
|
|
void SetValueName(Value value, const std::string name);
|
|
|
|
std::map<std::string, std::string> RenameValue(Value value,
|
|
const std::string &new_name,
|
|
Block *block);
|
|
std::optional<std::string> GetValueInputName(Value value);
|
|
|
|
std::vector<std::string> GetValueOutputNames(Value value);
|
|
Value GetOutputValueByName(const Program &program, const std::string &name);
|
|
Value GetValueByNameInPhiKernelProgram(const Program &program,
|
|
const std::string &name);
|
|
inline bool HasOnlyOneValueName(Value value) {
|
|
std::vector<std::string> names = GetValueAllNames(value);
|
|
return std::set<std::string>(names.begin(), names.end()).size() <= 1;
|
|
}
|
|
std::unordered_map<std::string, Value> GetAllNamedValues(
|
|
const Program &program);
|
|
|
|
} // namespace name_analysis
|
|
} // namespace utils
|
|
} // namespace pir
|