1 Star 0 Fork 146

镇亮/acl

forked from CANN/acl 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
dump.h 1.29 KB
一键复制 编辑 原始数据 按行查看 历史
liuxiangkun 提交于 2021-11-09 14:57 . char->char_t fix add
/**
* @file dump.h
*
* Copyright (C) Huawei Technologies Co., Ltd. 2019-2020. All Rights Reserved.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
*/
#ifndef ACL_DUMP_H_
#define ACL_DUMP_H_
#include "executor/ge_executor.h"
#include "nlohmann/json.hpp"
#include "acl/acl_base.h"
#include "common_inner.h"
namespace acl {
struct DumpInfo {
std::string modelName;
std::vector <std::string> layer;
bool isLayer = false; // Whether the label of "layer" exists
};
struct DumpConfig {
std::string dumpPath;
std::string dumpMode;
std::vector <DumpInfo> dumpList;
std::string dumpStatus;
std::string dumpOpSwitch;
};
class AclDump {
public:
static aclError HandleDumpConfig(const char_t *const configPath);
static AclDump &GetInstance();
void SetAclDumpFlag(const bool flag) { aclDumpFlag_ = flag; }
bool GetAclDumpFlag() const { return aclDumpFlag_; }
private:
static aclError HandleDumpCommand(const ge::DumpConfig &dumpCfg);
bool aclDumpFlag_ = false;
~AclDump() = default;
AclDump() = default;
};
}
#endif // ACL_DUMP_H_
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/Ronnie_zheng/acl.git
git@gitee.com:Ronnie_zheng/acl.git
Ronnie_zheng
acl
acl
master

搜索帮助