device_api.h File Reference

Abstract device memory management API. More...

#include <tvm/ffi/any.h>
#include <tvm/ffi/container/tensor.h>
#include <tvm/ffi/device.h>
#include <tvm/ffi/error.h>
#include <tvm/ffi/optional.h>
#include <tvm/ffi/string.h>
#include <tvm/runtime/base.h>
#include <string>

Include dependency graph for device_api.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::runtime::DeviceAPI
	TVM Runtime Device API, abstracts the device specific interface for memory management. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::runtime

Typedefs
typedef void *	TVMStreamHandle
	The stream that is specific to device can be NULL, which indicates the default one.
using	tvm::Device = DLDevice

Enumerations
enum	tvm::runtime::TVMDeviceExtType { tvm::runtime::TVMDeviceExtType_End = 36 }
	Extension device types in TVM. More...

Functions
const char *	tvm::runtime::DLDeviceType2Str (int type)
	The name of DLDeviceType.
std::ostream &	tvm::runtime::operator<< (std::ostream &os, DLDevice dev)
TVM_RUNTIME_DLL bool	tvm::runtime::RuntimeEnabled (const ffi::String &target)
	Check if runtime module is enabled for target.

Variables
constexpr int	tvm::runtime::kAllocAlignment = 64
	Number of bytes each allocation must align to.
constexpr int	tvm::runtime::kTempAllocaAlignment = 64
	Number of bytes each allocation must align to in temporary allocation.
constexpr int	tvm::runtime::kMaxStackAlloca = 1024
	Maximum size that can be allocated on stack.
constexpr int	tvm::runtime::kDefaultWorkspaceAlignment = 1
	Number of bytes each allocation must align to by default in the workspace buffer to service intermediate tensors.

Detailed Description

Abstract device memory management API.

Typedef Documentation

TVMStreamHandle

typedef void* TVMStreamHandle

The stream that is specific to device can be NULL, which indicates the default one.