Core C++ APIs - 3.5 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 English

wego_torch::core::Compile

Prototype

torch::jit::Module Compile(const torch::jit::Module &module,
                           CompileOptions options);
Table 1. Parameters
Type Name Description
const torch::jit::Module & module A quantized PyTorch module to be compiled, represented as a torch::jit::Module object.
CompileOptions options Compiler options used for WeGO-Torch compilation, specified as the wego_torch::core::CompileOptions data type. For further details on this object class, refer to the core classes section.

Return

An optimized torch::jit::Module object.

wego_torch::core::GetTargetInfo

Prototype

TargetInfo GetTargetInfo();

Parameters

None

Return

A wego_torch::TargetInfo object. See the classes section for more details about this object class type.

wego_torch::getVersionInfo

Prototype

std::string getVersionInfo();

Parameters

None

Return

A raw string representing wego_torch version information.