wehub-resource-sync
122 lines
3.4 KiB
C++
122 lines
3.4 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 "glog/logging.h"
|
|
#include "paddle/fluid/pir/dialect/distributed/ir/dist_dialect.h"
|
|
#include "paddle/fluid/pir/dialect/operator/ir/op_dialect.h"
|
|
#include "paddle/pir/include/core/builtin_dialect.h"
|
|
#include "paddle/pir/include/dialect/control_flow/ir/cf_dialect.h"
|
|
#include "paddle/utils/test_macros.h"
|
|
|
|
namespace pir {
|
|
/**
|
|
* IMPORTANT!!!
|
|
* all those defining strings can't be changed, otherwise the deserialization
|
|
* will failed. define all keys in serialized files to ensure accuracy for
|
|
* deserialization make sure all the key mutually exclusive
|
|
*/
|
|
|
|
// all IR structure's identifier (region, block, op, attr, type value etc)
|
|
// which can be string , int64_t etc.
|
|
#define ID "#"
|
|
#define VALUE_ID "%"
|
|
// program's key:
|
|
#define REGIONS "regions"
|
|
|
|
// region's key:
|
|
// which is json array with block json object(ID and BLOCKARGS and BLOCKOPS)
|
|
#define BLOCKS "blocks"
|
|
|
|
// block's key:
|
|
// which is json array with value json object
|
|
#define BLOCKARGS "args"
|
|
#define KEYWORDBLOCKARGS "kwargs"
|
|
|
|
#define KEYWORDNAME "key"
|
|
|
|
// which is json array with operation json object
|
|
#define BLOCKOPS "ops"
|
|
|
|
// operation's key:
|
|
// input
|
|
// which is json array with opoperand json object(ID)
|
|
#define OPOPERANDS "I"
|
|
|
|
// output
|
|
// which is json array with value json object(ID and TYPE_TYPE)
|
|
#define OPRESULTS "O"
|
|
|
|
// which is json array with json object(NAME and ATTR_TYPE)
|
|
#define ATTRS "A"
|
|
#define OPRESULTS_ATTRS "OA"
|
|
#define DIST_ATTRS "DA"
|
|
#define QUANT_ATTRS "QA"
|
|
|
|
// value's key:
|
|
// value's type which should be pir::Type's json object(ID or ID and DATA).
|
|
#define TYPE_TYPE "TT"
|
|
|
|
// attr's name which is operation's feature.
|
|
#define NAME "N"
|
|
|
|
// attr's value which is pir::Attribute's json object(ID and DATA).
|
|
#define ATTR_TYPE "AT"
|
|
|
|
// type/attr's contents which is json::array.
|
|
#define DATA "D"
|
|
// float/double data with nan, inf, -inf
|
|
#define VOID_DATA "VD"
|
|
|
|
// NULL_TYPE
|
|
#define NULL_TYPE "NULL"
|
|
|
|
// special op compress
|
|
#define PARAMETEROP "p"
|
|
|
|
// actions for patch
|
|
#define DELETE "DEL"
|
|
#define ADD "ADD"
|
|
#define UPDATE "UPD"
|
|
#define NEW_NAME "NN"
|
|
#define ADD_ATTRS "ADD_A"
|
|
#define ADD_OPRESULTS_ATTRS "ADD_OA"
|
|
#define PATCH "patch"
|
|
|
|
std::pair<std::string, std::string> GetContentSplitByDot(
|
|
const std::string& str);
|
|
|
|
std::vector<std::string> GetOpDistAttr();
|
|
std::vector<std::string> GetOpQuantAttr();
|
|
|
|
void GetCompressOpName(std::string* op_name);
|
|
|
|
void GetDecompressOpName(std::string* op_name);
|
|
class TEST_API DialectIdMap {
|
|
public:
|
|
static DialectIdMap* Instance();
|
|
DialectIdMap();
|
|
void insert(const std::string& key, const std::string& value);
|
|
|
|
std::string GetCompressDialectId(const std::string& name);
|
|
|
|
std::string GetDecompressDialectId(const std::string& id);
|
|
|
|
private:
|
|
std::unordered_map<std::string, std::string> CompressDialect;
|
|
std::unordered_map<std::string, std::string> DecompressDialect;
|
|
};
|
|
|
|
} // namespace pir
|