亲和库
来源介绍
针对开源模型中常见的网络结构和函数,我们针对性地对其进行了优化,使得运算性能大幅度提升,同时将其集成到PyTorch框架中,便于模型性能调优中使用。
功能介绍
当前框架中亲和函数可替换模型中一个或几个原生函数或几行关键代码,这取决与亲和函数的功能与模型中原生函数的功能。用户需明确模型内原生函数的功能,并根据提供的亲和函数替换。亲和库详细说明请参见API列表中“{对应的PyTorch版本号}>PyTorch Adapter自定义API>torch_npu.contrib”章节,此章节还提供了原生函数的代码、算法或参考论文地址。该部分调优内容会随着版本不断增强和更新,请以实际PyTorch版本中对应路径下的内容为准。
父主题: 参考信息