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