tvm
Public Attributes | List of all members
tvm::runtime::WorkerZeroData Class Reference

A special communication channel between controler and worker-0, assuming they are always collocated in the same process. More...

#include <session.h>

Collaboration diagram for tvm::runtime::WorkerZeroData:

Public Attributes

std::queue< NDArrayhost_arrays
 The host-side arrays to passed to worker-0 for special uses, for example, copy-to-worker0 and copy-from-worker0. More...
 
std::mutex queue_mutex_
 The mutex that guards host_arrays More...
 

Detailed Description

A special communication channel between controler and worker-0, assuming they are always collocated in the same process.

Member Data Documentation

◆ host_arrays

std::queue<NDArray> tvm::runtime::WorkerZeroData::host_arrays

The host-side arrays to passed to worker-0 for special uses, for example, copy-to-worker0 and copy-from-worker0.

◆ queue_mutex_

std::mutex tvm::runtime::WorkerZeroData::queue_mutex_

The mutex that guards host_arrays


The documentation for this class was generated from the following file: