本申請(qǐng)?zhí)峁┮环N張量計(jì)算代碼優(yōu)化方法、裝置、電子設(shè)備及介質(zhì)。其中,所述方法包括:分析張量計(jì)算代碼的循環(huán)特征和計(jì)算圖特征,得到相應(yīng)的循環(huán)信息和計(jì)算圖信息,根據(jù)所述循環(huán)信息和計(jì)算圖信息以及預(yù)設(shè)優(yōu)化方法,生成優(yōu)化空間,該優(yōu)化空間中每個(gè)空間點(diǎn)表示一個(gè)預(yù)設(shè)優(yōu)化方法組合及參數(shù)選擇。基于模擬退火算法和強(qiáng)化學(xué)習(xí)算法,在優(yōu)化空間中搜索并確定目標(biāo)空間點(diǎn),根據(jù)所述目標(biāo)空間點(diǎn)對(duì)應(yīng)的預(yù)設(shè)優(yōu)化方法組合及參數(shù)選擇,對(duì)所述張量計(jì)算代碼實(shí)施優(yōu)化,從而可以快速完成張量計(jì)算代碼的自動(dòng)優(yōu)化,提高張量計(jì)算代碼運(yùn)行效率,對(duì)于編程開發(fā)人員來說,可以免除開發(fā)算子的人力投入,并且取得相對(duì)較好的性能,可以減少成本并提高開發(fā)效率。
聲明:
“張量計(jì)算代碼優(yōu)化方法、裝置、設(shè)備及介質(zhì)” 該技術(shù)專利(論文)所有權(quán)利歸屬于技術(shù)(論文)所有人。僅供學(xué)習(xí)研究,如用于商業(yè)用途,請(qǐng)聯(lián)系該技術(shù)所有人。
我是此專利(論文)的發(fā)明人(作者)