28 #ifndef TVM_RUNTIME_C_BACKEND_API_H_ 29 #define TVM_RUNTIME_C_BACKEND_API_H_ 49 const char* func_name,
157 #endif // TVM_RUNTIME_C_BACKEND_API_H_ TVM_DLL int TVMBackendRegisterSystemLibSymbol(const char *name, void *ptr)
Backend function to register system-wide library symbol.
TVM_DLL int TVMBackendParallelLaunch(FTVMParallelLambda flambda, void *cdata, int num_task)
Backend function for running parallel jobs.
Environment for TVM parallel task.
TVM_DLL int TVMBackendRunOnce(void **handle, int(*f)(void *), void *cdata, int nbytes)
Simple static initialization function. Run f once and set handle to be not null. This function is mai...
TVM_DLL int TVMBackendFreeWorkspace(int device_type, int device_id, void *ptr)
Backend function to free temporal workspace.
void * TVMFunctionHandle
Handle to packed function handle.
TVM_DLL int TVMBackendParallelBarrier(int task_id, TVMParallelGroupEnv *penv)
BSP barrrier between parallel threads.
int(* FTVMParallelLambda)(int task_id, TVMParallelGroupEnv *penv, void *cdata)
The callback function to execute a parallel lambda.
void * sync_handle
Auxiliary used for synchronization.
TVM_DLL void * TVMBackendAllocWorkspace(int device_type, int device_id, uint64_t nbytes, int dtype_code_hint, int dtype_bits_hint)
Backend function to allocate temporal workspace.
TVM_DLL int TVMBackendGetFuncFromEnv(void *mod_node, const char *func_name, TVMFunctionHandle *out)
Backend function for modules to get function from its environment mod_node (its imports and global fu...
int32_t num_task
total amount of task