taoxiangdong
4 years ago
parent
1f05cd0b5c
commit
7370fb9c63
@ -0,0 +1,60 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||||
|
*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef AICPU_OP_TYPE_LIST_H_
|
||||||
|
#define AICPU_OP_TYPE_LIST_H_
|
||||||
|
|
||||||
|
enum OpKernelType {
|
||||||
|
TF_KERNEL,
|
||||||
|
CPU_KERNEL
|
||||||
|
};
|
||||||
|
|
||||||
|
enum ReturnCode {
|
||||||
|
OP_TYPE_NOT_SUPPORT,
|
||||||
|
FORMAT_NOT_SUPPORT,
|
||||||
|
DTYPE_NOT_SUPPORT
|
||||||
|
};
|
||||||
|
|
||||||
|
#pragma pack(push, 1)
|
||||||
|
//One byte alignment
|
||||||
|
struct SysOpInfo {
|
||||||
|
uint64_t opLen;
|
||||||
|
uint64_t opType;
|
||||||
|
OpKernelType kernelsType;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct OpParamInfo {
|
||||||
|
uint64_t num;
|
||||||
|
uint64_t dtypeList;
|
||||||
|
uint64_t formatList;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct SysOpCheckInfo {
|
||||||
|
uint64_t opListNum;
|
||||||
|
uint64_t offSetLen;
|
||||||
|
uint64_t sysOpInfoList;
|
||||||
|
uint64_t opParamInfoList;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct SysOpCheckResp {
|
||||||
|
uint64_t opListNum;
|
||||||
|
bool isWithoutJson;
|
||||||
|
uint64_t returnCodeList;
|
||||||
|
uint64_t sysOpInfoList;
|
||||||
|
uint64_t opParamInfoList;
|
||||||
|
};
|
||||||
|
#pragma pack(pop)
|
||||||
|
#endif // AICPU_OP_TYPE_LIST_H_
|
||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,135 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2020-2020 Huawei Technologies Co., Ltd
|
||||||
|
*
|
||||||
|
* 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.
|
||||||
|
*
|
||||||
|
* @file prof_callback.h
|
||||||
|
* @brief declaraion of profiling callbacks
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef MSPROFILER_PROF_CALLBACK_H_
|
||||||
|
#define MSPROFILER_PROF_CALLBACK_H_
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif // __cplusplus
|
||||||
|
|
||||||
|
|
||||||
|
#include "stddef.h"
|
||||||
|
#include "stdint.h"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofErrorCode
|
||||||
|
* @brief error code
|
||||||
|
*/
|
||||||
|
enum MsprofErrorCode {
|
||||||
|
MSPROF_ERROR_NONE = 0,
|
||||||
|
MSPROF_ERROR_MEM_NOT_ENOUGH,
|
||||||
|
MSPROF_ERROR_GET_ENV,
|
||||||
|
MSPROF_ERROR_CONFIG_INVALID,
|
||||||
|
MSPROF_ERROR_ACL_JSON_OFF,
|
||||||
|
MSPROF_ERROR,
|
||||||
|
};
|
||||||
|
|
||||||
|
#define MSPROF_ENGINE_MAX_TAG_LEN (31)
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name ReporterData
|
||||||
|
* @brief struct of data to report
|
||||||
|
*/
|
||||||
|
struct ReporterData {
|
||||||
|
char tag[MSPROF_ENGINE_MAX_TAG_LEN + 1]; // the sub-type of the module, data with different tag will be writen
|
||||||
|
int deviceId; // the index of device
|
||||||
|
size_t dataLen; // the length of send data
|
||||||
|
unsigned char *data; // the data content
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofReporterModuleId
|
||||||
|
* @brief module id of data to report
|
||||||
|
*/
|
||||||
|
enum MsprofReporterModuleId {
|
||||||
|
MSPROF_MODULE_DATA_PREPROCESS = 0, // DATA_PREPROCESS
|
||||||
|
MSPROF_MODULE_HCCL, // HCCL
|
||||||
|
MSPROF_MODULE_ACL, // AclModule
|
||||||
|
MSPROF_MODULE_FRAMEWORK, // Framework
|
||||||
|
MSPROF_MODULE_RUNTIME // runtime
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofReporterCallbackType
|
||||||
|
* @brief reporter callback request type
|
||||||
|
*/
|
||||||
|
enum MsprofReporterCallbackType {
|
||||||
|
MSPROF_REPORTER_REPORT = 0, // report data
|
||||||
|
MSPROF_REPORTER_INIT, // init reporter
|
||||||
|
MSPROF_REPORTER_UNINIT, // uninit reporter
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofReporterCallback
|
||||||
|
* @brief callback to start reporter/stop reporter/report date
|
||||||
|
* @param moduleId [IN] enum MsprofReporterModuleId
|
||||||
|
* @param type [IN] enum MsprofReporterCallbackType
|
||||||
|
* @param data [IN] callback data (nullptr on INTI/UNINIT)
|
||||||
|
* @param len [IN] callback data size (0 on INIT/UNINIT)
|
||||||
|
* @return enum MsprofErrorCode
|
||||||
|
*/
|
||||||
|
typedef int32_t (*MsprofReporterCallback)(uint32_t moduleId, uint32_t type, void *data, uint32_t len);
|
||||||
|
|
||||||
|
|
||||||
|
#define MSPROF_OPTIONS_DEF_LEN_MAX (2048)
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofGeOptions
|
||||||
|
* @brief struct of MSPROF_CTRL_INIT_GE_OPTIONS
|
||||||
|
*/
|
||||||
|
struct MsprofGeOptions {
|
||||||
|
char jobId[MSPROF_OPTIONS_DEF_LEN_MAX];
|
||||||
|
char options[MSPROF_OPTIONS_DEF_LEN_MAX];
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofCtrlCallbackType
|
||||||
|
* @brief ctrl callback request type
|
||||||
|
*/
|
||||||
|
enum MsprofCtrlCallbackType {
|
||||||
|
MSPROF_CTRL_INIT_ACL_ENV = 0, // start profiling with acl env
|
||||||
|
MSPROF_CTRL_INIT_ACL_JSON, // start profiling with acl.json
|
||||||
|
MSPROF_CTRL_INIT_GE_OPTIONS, // start profiling with ge env and options
|
||||||
|
MSPROF_CTRL_FINALIZE // stop profiling
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofCtrlCallback
|
||||||
|
* @brief callback to start/stop profiling
|
||||||
|
* @param type [IN] enum MsprofCtrlCallbackType
|
||||||
|
* @param data [IN] callback data
|
||||||
|
* @param len [IN] callback data size
|
||||||
|
* @return enum MsprofErrorCode
|
||||||
|
*/
|
||||||
|
typedef int32_t (*MsprofCtrlCallback)(uint32_t type, void *data, uint32_t len);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @name MsprofSetDeviceCallback
|
||||||
|
* @brief callback to notify set/reset device
|
||||||
|
* @param devId [IN] device id
|
||||||
|
* @param isOpenDevice [IN] true: set device, false: reset device
|
||||||
|
*/
|
||||||
|
typedef void (*MsprofSetDeviceCallback)(uint32_t devId, bool isOpenDevice);
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif // MSPROFILER_PROF_CALLBACK_H_
|
||||||
Loading…
Reference in new issue