m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update thirdparty includes

Browse Source
pull/234/head
taoxiangdong 5 years ago
parent 11851b69bf
commit ce6b049eba
17 changed files with 627 additions and 394 deletions
Show Stats Download Patch File Download Diff File

6
third_party/fwkacllib/inc/hccl/base.h vendored
Unescape View File

@ -68,8 +68,10 @@ struct MemRegisterAddr {
u64 addr;
u64 length;
};
const u32 HCCL_MAX_MEM_REGISTER_NUM = 8; // The max number of memory register address.
/*
* @brief The max number of memory register address for remote access.
*/
const u32 HCCL_MAX_MEM_REGISTER_NUM = 32;
enum GradSplitForceMode {
FORCE_NONE, /**< no force */

13
third_party/fwkacllib/inc/mmpa/mmpa_api.h vendored
Unescape View File

@ -26,6 +26,12 @@
#define _GNU_SOURCE
#endif
#ifdef FUNC_VISIBILITY
#define MMPA_FUNC_VISIBILITY __attribute__((visibility("default")))
#else
#define MMPA_FUNC_VISIBILITY
#endif
#include <string.h>
#include <stdio.h>
#include <errno.h>
@ -86,6 +92,13 @@
#if(OS_TYPE == WIN) //lint !e553
#ifdef FUNC_VISIBILITY
#define MMPA_FUNC_VISIBILITY _declspec(dllexport)
#else
#define MMPA_FUNC_VISIBILITY
#endif
#include <winsock2.h>
#include <winsock.h>
#include "Windows.h"

352
third_party/fwkacllib/inc/mmpa/sub_inc/mmpa_linux.h vendored
View File

File diff suppressed because it is too large Load Diff

366
third_party/fwkacllib/inc/mmpa/sub_inc/mmpa_win.h vendored
View File

File diff suppressed because it is too large Load Diff

16
third_party/fwkacllib/inc/runtime/base.h vendored
Unescape View File

@ -391,6 +391,8 @@ typedef void (*rtErrorCallback)(rtExceptionType);
typedef void (*rtTaskFailCallback)(rtExceptionInfo *exceptionInfo);
typedef void (*rtDeviceStateCallback)(uint32_t devId, bool isOpen);
/**
* @ingroup dvrt_base
* @brief stream handle.
@ -419,7 +421,7 @@ RTS_API rtError_t rtSetProfDirEx(const char *profDir, const char *address, const
* @ingroup profiling_base
* @brief init profiler object.
*/
RTS_API rtError_t rtProfilerInit(const char *profdir, const char *address, const char *job_ctx);
RTS_API rtError_t rtProfilerInit(const char *profDir, const char *address, const char *jobCtx);
/**
* @ingroup profiling_base
@ -473,6 +475,16 @@ RTS_API rtError_t rtSetExceptCallback(rtErrorCallback callback);
*/
RTS_API rtError_t rtSetTaskFailCallback(rtTaskFailCallback callback);
/**
* @ingroup dvrt_base
* @brief register callback for deviceid
* @param [in] uniName unique register name, can't be null
* @param [in] callback Device state callback function
* @param [out] NA
* @return RT_ERROR_NONE for ok
*/
RTS_API rtError_t rtRegDeviceStateCallback(const char *regName, rtDeviceStateCallback callback);
/**
* @ingroup dvrt_base
* @brief notify handle.
@ -593,7 +605,7 @@ RTS_API rtError_t rtLabelCreateEx(rtLabel_t *label, rtStream_t stream);
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for input null ptr
*/
RTS_API rtError_t rtGetTaskIdAndStreamID(uint32_t *taskid, uint32_t *streamid);
RTS_API rtError_t rtGetTaskIdAndStreamID(uint32_t *taskId, uint32_t *streamId);
#if defined(__cplusplus) && !defined(COMPILE_OMG_PACKAGE)
}

2
third_party/fwkacllib/inc/runtime/context.h vendored
Unescape View File

@ -139,7 +139,7 @@ RTS_API rtError_t rtSetGroup(int32_t groupId);
* @param [in] groupid count
* @return RT_ERROR_NONE for ok, errno for failed
*/
RTS_API rtError_t rtGetGroupInfo(int32_t groupId, rtGroupInfo_t* groupInfo, uint32_t count);
RTS_API rtError_t rtGetGroupInfo(int32_t groupId, rtGroupInfo_t *groupInfo, uint32_t count);
/**
* @ingroup

4
third_party/fwkacllib/inc/runtime/dev.h vendored
Unescape View File

@ -295,7 +295,7 @@ RTS_API rtError_t rtGetRunMode(rtRunMode *mode);
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_DRV_ERR for can not get aicpu deploy
*/
RTS_API rtError_t rtGetAicpuDeploy(rtAicpuDeployType_t *deplyType);
RTS_API rtError_t rtGetAicpuDeploy(rtAicpuDeployType_t *deployType);
/**
* @ingroup dvrt_dev
@ -309,7 +309,7 @@ RTS_API rtError_t rtSetSocVersion(const char *version);
* @brief get chipType
* @return RT_ERROR_NONE for ok
*/
rtError_t rtGetSocVersion(char *version, const uint32_t maxLen);
RTS_API rtError_t rtGetSocVersion(char *version, const uint32_t maxLen);
/**
* @ingroup dvrt_dev

28
third_party/fwkacllib/inc/runtime/event.h vendored
Unescape View File

@ -57,6 +57,16 @@ RTS_API rtError_t rtEventCreateWithFlag(rtEvent_t *event, uint32_t flag);
*/
RTS_API rtError_t rtEventDestroy(rtEvent_t event);
/**
* @ingroup dvrt_event
* @brief get event id
* @param [in] event_ event to be get
* @param [in|out] event_id event_id id
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtGetEventID(rtEvent_t event, uint32_t *eventId);
/**
* @ingroup dvrt_event
* @brief event record
@ -116,13 +126,13 @@ RTS_API rtError_t rtEventGetTimeStamp(uint64_t *time, rtEvent_t event);
/**
* @ingroup dvrt_event
* @brief name an event
* @param [in] event_ event to be named
* @param [in] event event to be named
* @param [in] name identification name
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input of event, name
* @return RT_ERROR_DRV_ERR for driver error
*/
RTS_API rtError_t rtNameEvent(rtEvent_t event_, const char *name);
RTS_API rtError_t rtNameEvent(rtEvent_t event, const char *name);
/**
* @ingroup dvrt_event
@ -132,7 +142,7 @@ RTS_API rtError_t rtNameEvent(rtEvent_t event_, const char *name);
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtNotifyCreate(int32_t device_id, rtNotify_t *notify_);
RTS_API rtError_t rtNotifyCreate(int32_t deviceId, rtNotify_t *notify);
/**
* @ingroup dvrt_event
@ -142,7 +152,7 @@ RTS_API rtError_t rtNotifyCreate(int32_t device_id, rtNotify_t *notify_);
* @return RT_ERROR_INVALID_VALUE for error input
* @return RT_ERROR_DRV_ERR for driver error
*/
RTS_API rtError_t rtNotifyDestroy(rtNotify_t notify_);
RTS_API rtError_t rtNotifyDestroy(rtNotify_t notify);
/**
* @ingroup dvrt_event
@ -153,7 +163,7 @@ RTS_API rtError_t rtNotifyDestroy(rtNotify_t notify_);
* @return RT_ERROR_INVALID_VALUE for error input
* @return RT_ERROR_STREAM_CONTEXT for stream is not in current ctx
*/
RTS_API rtError_t rtNotifyRecord(rtNotify_t notify_, rtStream_t stream_);
RTS_API rtError_t rtNotifyRecord(rtNotify_t notify, rtStream_t stream);
/**
* @ingroup dvrt_event
@ -164,7 +174,7 @@ RTS_API rtError_t rtNotifyRecord(rtNotify_t notify_, rtStream_t stream_);
* @return RT_ERROR_INVALID_VALUE for error input
* @return RT_ERROR_STREAM_CONTEXT for stream is not in current ctx
*/
RTS_API rtError_t rtNotifyWait(rtNotify_t notify_, rtStream_t stream_);
RTS_API rtError_t rtNotifyWait(rtNotify_t notify, rtStream_t stream);
/**
* @ingroup dvrt_event
@ -174,7 +184,7 @@ RTS_API rtError_t rtNotifyWait(rtNotify_t notify_, rtStream_t stream_);
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtNameNotify(rtNotify_t notify_, const char *name);
RTS_API rtError_t rtNameNotify(rtNotify_t notify, const char *name);
/**
* @ingroup dvrt_event
@ -184,7 +194,7 @@ RTS_API rtError_t rtNameNotify(rtNotify_t notify_, const char *name);
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtGetNotifyID(rtNotify_t notify_, uint32_t *notify_id);
RTS_API rtError_t rtGetNotifyID(rtNotify_t notify, uint32_t *notifyId);
/**
* @ingroup dvrt_event
@ -199,7 +209,7 @@ RTS_API rtError_t rtIpcSetNotifyName(rtNotify_t notify, char *name, uint32_t len
/**
* @ingroup dvrt_event
* @brief Open IPC notify
* @param [in] notify notify to be opened
* @param [out] notify the opened notify
* @param [in] name identification name
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input

33
third_party/fwkacllib/inc/runtime/kernel.h vendored
Unescape View File

@ -169,6 +169,18 @@ typedef void (*rtCallback_t)(void *fnData);
*/
#define RT_DEV_BINARY_MAGIC_ELF_AIVEC 0x41415246
/**
* @ingroup rt_kernel
* @brief magic number of elf binary for aicube
*/
#define RT_DEV_BINARY_MAGIC_ELF_AICUBE 0x41415247
/**
* @ingroup rt_kernel
* @brief magic number of elf binary for aivector
*/
#define RT_DEV_BINARY_MAGIC_ELF_AIVECTOR 0x41415248
/**
* @ingroup rt_kernel_flags
* @brief kernel op bit flags
@ -285,7 +297,7 @@ RTS_API rtError_t rtQueryFunctionRegistered(const char *stubName);
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtKernelConfigDump(uint32_t kind, uint32_t dumpSizePerBlock, uint32_t blockDim, void **dumpBaseAddr,
rtStream_t stream_);
rtStream_t stream);
/**
* @ingroup rt_kernel
@ -375,7 +387,7 @@ typedef void *rtModel_t;
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtDumpAddrSet(rtModel_t model, void *addr , uint32_t dumpSize, uint32_t flag);
RTS_API rtError_t rtDumpAddrSet(rtModel_t model, void *addr, uint32_t dumpSize, uint32_t flag);
/**
* @ingroup rt_kernel
@ -529,6 +541,23 @@ RTS_API rtError_t rtStopOnlineProf(rtStream_t stream);
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtGetOnlineProfData(rtStream_t stream, rtProfDataInfo_t *pProfData, uint32_t profDataNum);
/**
* @ingroup profiling_base
* @brief start mdc profiler.
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtStartMDCProfiler(void **addr, uint32_t length);
/**
* @ingroup profiling_base
* @brief stop mdc profiler.
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtStopMDCProfiler(void *addr);
#if defined(__cplusplus) && !defined(COMPILE_OMG_PACKAGE)
}
#endif

6
third_party/fwkacllib/inc/runtime/mem.h vendored
Unescape View File

@ -475,7 +475,7 @@ RTS_API rtError_t rtIpcSetMemoryName(const void *ptr, uint64_t byteCount, char *
* @return RT_ERROR_INVALID_VALUE for error input
* @return RT_ERROR_DRV_ERR for driver error
*/
rtError_t rtIpcDestroyMemoryName(const char *name);
RTS_API rtError_t rtIpcDestroyMemoryName(const char *name);
/**
* @ingroup dvrt_mem
@ -503,13 +503,13 @@ RTS_API rtError_t rtIpcCloseMemory(const void *ptr);
* @ingroup dvrt_mem
* @brief HCCL Async memory cpy
* @param [in] index sq index
* @param [in] wqe_index moudle index
* @param [in] wqeIndex moudle index
* @param [in] stream asynchronized task stream
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
* @return RT_ERROR_DRV_ERR for driver error
*/
RTS_API rtError_t rtRDMASend(uint32_t index, uint32_t wqe_index, rtStream_t stream);
RTS_API rtError_t rtRDMASend(uint32_t index, uint32_t wqeIndex, rtStream_t stream);
/**
* @ingroup dvrt_mem

3
third_party/fwkacllib/inc/runtime/rt_model.h vendored
Unescape View File

@ -438,7 +438,8 @@ RTS_API rtError_t rtModelGetId(rtModel_t model, uint32_t *modelId);
* @return RT_ERROR_NONE for ok
* @return RT_ERROR_INVALID_VALUE for error input
*/
rtError_t rtDebugRegister(rtModel_t model, uint32_t flag, const void *addr, uint32_t *streamId, uint32_t *taskId);
RTS_API rtError_t rtDebugRegister(rtModel_t model, uint32_t flag, const void *addr,
uint32_t *streamId, uint32_t *taskId);
/*
* @ingroup rt_model

16
third_party/fwkacllib/inc/runtime/stream.h vendored
Unescape View File

@ -126,17 +126,17 @@ RTS_API rtError_t rtGetStreamId(rtStream_t stream, int32_t *streamId);
* @return RT_ERROR_NONE for complete
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtGetMaxStreamAndTask(uint32_t streamType, uint32_t *MaxStrCount, uint32_t *MaxTaskCount);
RTS_API rtError_t rtGetMaxStreamAndTask(uint32_t streamType, uint32_t *maxStrCount, uint32_t *maxTaskCount);
/**
* @ingroup dvrt_stream
* @brief Name a stream
* @param [in] stream_ stream to be named
* @param [in] stream stream to be named
* @param [in] name identification name
* @return RT_ERROR_NONE for complete
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtNameStream(rtStream_t stream_, const char *name);
RTS_API rtError_t rtNameStream(rtStream_t stream, const char *name);
/**
* @ingroup dvrt_stream
@ -144,12 +144,12 @@ RTS_API rtError_t rtNameStream(rtStream_t stream_, const char *name);
* @param [in] ptr Determine the address where the value of the true and false branches is located
* @param [in] condition switch condition
* @param [in] value switch value
* @param [in] true_stream Stream that needs to be activated when the value is non-zero
* @param [in] trueStream Stream that needs to be activated when the value is non-zero
* @param [in] stream input stream to init task
* @return RT_ERROR_NONE for complete
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtStreamSwitch(void *ptr, rtCondition_t condition, int64_t value, rtStream_t true_stream,
RTS_API rtError_t rtStreamSwitch(void *ptr, rtCondition_t condition, int64_t value, rtStream_t trueStream,
rtStream_t stream);
/**
@ -162,18 +162,18 @@ RTS_API rtError_t rtStreamSwitch(void *ptr, rtCondition_t condition, int64_t val
* @param [in] dataType data type of target value
* @return RT_ERROR_NONE for complete
*/
RTS_API rtError_t rtStreamSwitchEx(void *ptr, rtCondition_t condition, void *value_ptr, rtStream_t true_stream,
RTS_API rtError_t rtStreamSwitchEx(void *ptr, rtCondition_t condition, void *valuePtr, rtStream_t trueStream,
rtStream_t stream, rtSwitchDataType_t dataType);
/**
* @ingroup dvrt_stream
* @brief Active a stream
* @param [in] active_stream stream to be activated
* @param [in] activeStream stream to be activated
* @param [in] stream input stream to init task
* @return RT_ERROR_NONE for complete
* @return RT_ERROR_INVALID_VALUE for error input
*/
RTS_API rtError_t rtStreamActive(rtStream_t active_stream, rtStream_t stream);
RTS_API rtError_t rtStreamActive(rtStream_t activeStream, rtStream_t stream);
/**
* @brief execute extensible stream case switch task

29
third_party/fwkacllib/inc/tdt/index_transform.h vendored
Unescape View File

@ -0,0 +1,29 @@
/**
* @file index_transform.h
*
* Copyright (C) Huawei Technologies Co., Ltd. 2018-2019. All Rights Reserved.
*
* This program is used to get logical device id by phy device id.
*/
#ifndef INC_TDT_INDEX_TRANSFORM_H
#define INC_TDT_INDEX_TRANSFORM_H
#include "stdint.h"
/**
* @ingroup IndexTransform
* @brief get logical device id by phy device id.
*
* @par Function get logical device id by phy device id.
*
* @param phyId [IN] physical device id
* @param logicalId [OUT] logical device id
* @retval 0 Success
* @retval OtherValues Fail
*
* @par Dependency
* @li libruntime.so: Library to which the interface belongs.
*/
int32_t IndexTransform(const uint32_t phyId, uint32_t &logicId);
#endif

3
third_party/fwkacllib/inc/tdt/status.h vendored
Unescape View File

@ -193,6 +193,7 @@ enum {
TDT_HDC_SRV_TYPE_ERROR_CODE,
TDT_TSD_CLT_OPEN_FAILED_CODE,
TDT_TSD_CLT_CLOSE_FAILED_CODE,
TDT_TSD_CLT_UPDATE_PROFILING_FAILED_CODE,
TDT_TSD_CLT_INTERFACE_NOT_SUPPORT_CODE,
TDT_SUPERVISOR_ILLEGAL_HEARTBEAT_TIME_CODE,
TDT_SUPERVISOR_INOTIFY_READ_SIZE_ERROR_CODE,
@ -697,6 +698,8 @@ TDT_DEF_ERROR_CODE(MODID_HDC_SERVER, TDT_ERROR, TDT_BIND_CPUCORE_FAILED, "thread
TDT_DEF_ERROR_CODE(MODID_HDC_SERVER, TDT_ERROR, TDT_HDC_SRV_CLOSED, "hdc server has been closed");
TDT_DEF_ERROR_CODE(MODID_TSD_CLIENT, TDT_ERROR, TDT_TSD_CLT_OPEN_FAILED, "tsd client open failed");
TDT_DEF_ERROR_CODE(MODID_TSD_CLIENT, TDT_ERROR, TDT_TSD_CLT_CLOSE_FAILED, "tsd client close failed");
TDT_DEF_ERROR_CODE(MODID_TSD_CLIENT, TDT_ERROR, TDT_TSD_CLT_UPDATE_PROFILING_FAILED,
"tsd client update profiling failed");
TDT_DEF_ERROR_CODE(MODID_TSD_CLIENT, TDT_ERROR, TDT_TSD_CLT_INTERFACE_NOT_SUPPORT, "tsd client func not support");
TDT_DEF_ERROR_CODE(MODID_TDT_PREFETCH, TDT_ERROR, TDT_PREFETCH_FILELIST_NOT_EXIST, "tdt filelist open failed");
TDT_DEF_ERROR_CODE(MODID_TDT_PREFETCH, TDT_ERROR, TDT_PREFETCH_SAMPLE_FILE_NOT_FOUND, "tdt sample file is empty");

22
third_party/fwkacllib/inc/tdt/tsd_client.h vendored
Unescape View File

@ -49,7 +49,7 @@ extern "C" {
* @li tsd_client.h: Header file where the interface declaration is located.
* @li data_common.h: Header file where 'TDT_StatusT' defined
*/
TDT_StatusT TsdOpen(const uint32_t phyDeviceId, const uint32_t rankSize);
TDT_LIB_EXPORT TDT_StatusT TsdOpen(const uint32_t phyDeviceId, const uint32_t rankSize);
/**
* @ingroup Close
@ -67,7 +67,25 @@ TDT_StatusT TsdOpen(const uint32_t phyDeviceId, const uint32_t rankSize);
* @li tsd_client.h: Header file where the interface declaration is located.
* @li data_common.h: Header file where 'TDT_StatusT' defined
*/
TDT_StatusT TsdClose(const uint32_t phyDeviceId);
TDT_LIB_EXPORT TDT_StatusT TsdClose(const uint32_t phyDeviceId);
/**
* @ingroup UpdateProfilingMode
* @brief notify TSDClient update profiling mode
*
* @par Function
* notify TSDClient update profiling mode
*
* @param NA
* @retval TDT_OK Success
* @retval OtherValues Failure
*
* @par Dependency
* @li libtsdclient.so: Library to which the interface belongs.
* @li tsd_client.h: Header file where the interface declaration is located.
* @li data_common.h: Header file where 'TDT_StatusT' defined
*/
TDT_LIB_EXPORT TDT_StatusT UpdateProfilingMode(const uint32_t phyDeviceId, const uint32_t flag);
/**
* @ingroup CreateCmdParameterObj

26
third_party/fwkacllib/inc/toolchain/slog.h vendored
Unescape View File

@ -25,10 +25,20 @@ extern "C" {
#define LINUX 0
#endif // LINUX
#ifndef WIN
#define WIN 1
#endif
#ifndef OS_TYPE
#define OS_TYPE 0
#endif // OS_TYPE
#if (OS_TYPE == LINUX)
#define DLL_EXPORT __attribute__((visibility("default")))
#else
#define DLL_EXPORT _declspec(dllexport)
#endif
/**
* @ingroup slog
*
@ -180,12 +190,11 @@ enum {
INVLID_MOUDLE_ID
};
#if (OS_TYPE == LINUX)
/**
* @ingroup slog
* @brief External log interface, which called by modules
*/
extern void dlog_init(void);
DLL_EXPORT void dlog_init(void);
/**
* @ingroup slog
@ -195,7 +204,7 @@ extern void dlog_init(void);
* @param [out]enableEvent: 1: enable; 0: disable
* @return: module level(0: debug, 1: info, 2: warning, 3: error, 4: null output)
*/
extern int dlog_getlevel(int moduleId, int *enableEvent);
DLL_EXPORT int dlog_getlevel(int moduleId, int *enableEvent);
/**
* @ingroup slog
@ -206,7 +215,7 @@ extern int dlog_getlevel(int moduleId, int *enableEvent);
* @param [in]enableEvent: 1: enable; 0: disable, others:invalid
* @return: 0: SUCCEED, others: FAILED
*/
extern int dlog_setlevel(int moduleId, int level, int enableEvent);
DLL_EXPORT int dlog_setlevel(int moduleId, int level, int enableEvent);
/**
* @ingroup slog
@ -217,7 +226,7 @@ extern int dlog_setlevel(int moduleId, int level, int enableEvent);
* @param [in]logLevel: eg: DLOG_EVENT/DLOG_ERROR/DLOG_WARN/DLOG_INFO/DLOG_DEBUG
* @return: 1:enable, 0:disable
*/
extern int CheckLogLevel(int moduleId, int logLevel);
DLL_EXPORT int CheckLogLevel(int moduleId, int logLevel);
/**
* @ingroup slog
@ -343,7 +352,7 @@ extern int CheckLogLevel(int moduleId, int logLevel);
* @ingroup slog
* @brief DlogFlush: flush log buffer to file
*/
void DlogFlush(void);
DLL_EXPORT void DlogFlush(void);
/**
* @ingroup slog
@ -357,11 +366,6 @@ void DlogEventInner(int moduleId, const char *fmt, ...);
void DlogInner(int moduleId, int level, const char *fmt, ...);
void DlogWithKVInner(int moduleId, int level, KeyValue *pstKVArray, int kvNum, const char *fmt, ...);
#else
_declspec(dllexport) void dlog_init(void);
_declspec(dllexport) int dlog_getlevel(int moduleId, int *enableEvent);
#endif // OS_TYPE
#ifdef __cplusplus
}
#endif // __cplusplus

72
third_party/fwkacllib/inc/toolchain/tuning_tool/tune_api.h vendored
Unescape View File

@ -0,0 +1,72 @@
/**
* @file tune_api.h
*
* Copyright (c) Huawei Technologies Co., Ltd. 2020-2020. All rights reserved.\n
*
* 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.\n
* 描述：mstune调优接口头文件
*/
/** @defgroup mstune mstune调优接口 */
#ifndef TUNE_API_H
#define TUNE_API_H
#include <vector>
#include <map>
#include <string>
#include "graph/graph.h"
#include "ge/ge_api.h"
/**
* @ingroup mstune
*
* mstune status
*/
enum MsTuneStatus {
MSTUNE_SUCCESS, /** tune success */
MSTUNE_FAILED, /** tune failed */
};
// Option key: for train options sets
const std::string MSTUNE_SELF_KEY = "mstune";
const std::string MSTUNE_GEINIT_KEY = "initialize";
const std::string MSTUNE_GESESS_KEY = "session";
/**
* @ingroup mstune
* @par 描述: 命令行调优
*
* @attention 无
* @param option [IN] 调优参数
* @param msg [OUT] 调优异常下返回信息
* @retval #MSTUNE_SUCCESS 执行成功
* @retval #MSTUNE_FAILED 执行失败
* @par 依赖:
* @li tune_api.cpp：该接口所属的开发包。
* @li tune_api.h：该接口声明所在的头文件。
* @see 无
* @since
*/
MsTuneStatus MsTuning(const std::map<std::string, std::string> &option, std::string &msg);
/**
* @ingroup mstune
* @par 描述: 梯度调优
*
* @attention 无
* @param tuningGraph [IN] 调优图
* @param dependGraph [IN] 调优依赖图
* @param session [IN] ge连接会话
* @param option [IN] 参数集. 包含调优参数及ge参数
* @retval #MSTUNE_SUCCESS 执行成功
* @retval #MSTUNE_FAILED 执行失败
* @par 依赖:
* @li tune_api.cpp：该接口所属的开发包。
* @li tune_api.h：该接口声明所在的头文件。
* @see 无
* @since
*/
extern "C" MsTuneStatus MsTrainTuning(ge::Graph &tuningGraph, std::vector<ge::Graph> &dependGraph,
ge::Session *session, const std::map<std::string, std::map<std::string, std::string>> &option);
#endif
Write Preview
Loading…
Cancel
Save