WeGO-PyTorch Limitations - 2.5 English

Vitis AI User Guide (UG1414)
Document ID
UG1414
Release Date
2022-06-15
Version
2.5 English
VAI 2.5 WeGO-Torch is in early access state and has several limitations while using WeGO-Torch:
  1. Currently the optimized TorchScript module generated by WeGO-Torch cannot be serialized, which means you cannot use torch.jit.save to save the optimized TorchScript module.
  2. WeGO-Torch currently only covers a subset of operators that cloud DPUs can support, which means WeGO-Torch will dispatch some operators into CPU for execution even if these operators can be supported by Cloud DPUs.