tvm
Classes | Namespaces
compilation_config.h File Reference

A helper class to collect all the targets in canonical form necessary for compilation. More...

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

Go to the source code of this file.

Classes

class  tvm::CompilationConfigNode
 Gathers the Targets and distinguished VirtualDevices in canonical form needed to compile a Relay module for execution over possibly heterogeneous devices. Centralizes the validation and canonicalization logic needed to transition from targets supplied by the Python APIs to a single internal representation. Also holds a cache of canonical VirtualDevices so that structural equal virtual devices have pointer equal canonical virtual devices. More...
 
class  tvm::CompilationConfig
 Managed reference class to CompilationConfig. More...
 

Namespaces

 tvm
 runtime implementation for LibTorch/TorchScript.
 

Detailed Description

A helper class to collect all the targets in canonical form necessary for compilation.