tvm
Classes | Namespaces | Variables
virtual_device.h File Reference

A compile time representation for where data is to be stored at runtime, and how to compile code to compute it. More...

#include <tvm/ir/transform.h>
#include <tvm/target/target.h>
#include <string>
#include <unordered_set>
#include <utility>
Include dependency graph for virtual_device.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::VirtualDeviceNode
 Describes at compile time the constraints on where data is to be stored at runtime down to the (virtual) device and memory scope level, and how to compile code to compute that data. Used by the PlanDevices pass to collect and solve (virtual) device constraints for the whole Relay program. More...
 
class  tvm::VirtualDevice
 Managed reference class to VirtualDeviceNode. More...
 
class  tvm::VirtualDeviceCache
 A cache of VirtualDevices. This can be used: More...
 

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 

Variables

constexpr int tvm::kNullDeviceType = 0
 
constexpr int tvm::kInvalidDeviceType = -1
 
constexpr const char * tvm::kVirtualDevice = "virtual_device"
 

Detailed Description

A compile time representation for where data is to be stored at runtime, and how to compile code to compute it.