概述

本章节包含适配后的jit接口,提供即时编译能力以获取更好的性能及部署能力。

表1 torch_npu jit API

API名称

说明

(beta)torch_npu.jit.optimize

实现jit_mod优化。