torch.compile

训练和推理后端

torch.compile(m, backend="inductor")
torch.compile(m, backend="cudagraphs")
torch.compile(m, backend="ipex")

仅推理后端

torch.compile(m, backend="tensorrt")
torch.compile(m, backend="ipex")
torch.compile(m, backend="tvm")
torch.compile(m, backend="openvino")