一種AI模型編譯方法、設備及存儲介質(zhì)

基本信息

申請?zhí)?/td> CN201911040449.X 申請日 -
公開（公告）號 CN110750298B 公開（公告）日 2021-03-09
申請公布號 CN110750298B 申請公布日 2021-03-09
分類號 G06F8/76(2018.01)I;G06F8/41(2018.01)I 分類 計算；推算；計數(shù);
發(fā)明人 夏正勛;羅圣美 申請（專利權(quán)）人 南京星環(huán)智能科技有限公司
代理機構(gòu) 北京品源專利代理有限公司 代理人 孟金喆
地址 210019江蘇省南京市建鄴區(qū)賢坤路1號科創(chuàng)中心2樓220-351號
法律狀態(tài) -

摘要

摘要 本發(fā)明實施例公開了一種AI模型編譯方法、設備及存儲介質(zhì)。其中，方法包括：獲取通過標準AI框架訓練得到的AI模型，并解析AI模型；對解析得到的AI模型中的算子進行分類，得到待適配的AI芯片支持的第一類算子集以及AI芯片不支持的第二類算子集；根據(jù)第一類算子集以及第二類算子集，生成至少一項中間表達；根據(jù)至少一項中間表達生成用于在所述AI芯片上運行的執(zhí)行模型。本發(fā)明實施例的方案實現(xiàn)了將AI模型適配于不同的AI芯片，簡化了AI模型的應用。??