embedl_deploy.vitis package#

Subpackages:

Module contents:

Vitis backend — curated pattern lists and convenience API.

Quick start:

import torch
from torchvision.models import resnet50
from embedl_deploy import transform
from embedl_deploy.vitis import VITIS_PATTERNS

model = resnet50(weights=None).eval()
args = (torch.randn(1, 3, 224, 224),)
deployed = transform(model, args, patterns=VITIS_PATTERNS).model

Pattern lists#

VITIS_CONVERSION_PATTERNS

Structural conversions applied before fusion.

VITIS_FUSION_PATTERNS

Fusion-only patterns (Conv→BN→ReLU, etc.).

VITIS_PATTERNS

Union of conversion + fusion patterns for compilation with AMD Vitis AI. This is the recommended list for most users.

VITIS_QUANTIZED_PATTERNS

Q/DQ stub insertion patterns for quantization.

VITIS_SMOOTH_PATTERNS

SmoothQuant preparation patterns.