排序算子的编译方法和装置制造方法及图纸

技术编号:41881814 阅读:19 留言:0更新日期:2024-07-02 00:36
本申请公开了一种排序算子的编译方法和装置,该方法包括:接收用户输入的排序参数和选择调用的第一原语;其中,排序参数和第一原语用于对多维数据进行排序;根据排序参数和第一原语生成排序算子的调度策略;对排序算子的计算描述和调度策略进行编译,得到包含调度策略的排序计算表达。通过本申请,可以提升编译后排序算子的执行效率,提升排序算子的性能。

【技术实现步骤摘要】

本申请涉及信息,尤其涉及一种排序算子的编译方法和装置


技术介绍

1、随着人工智能技术的高速发展,传统的图形处理器(graphics processing unit,gpu)、中央处理器(central processing unit,cpu)已经不能满足日益增长的性能需求,各大厂商都在试图研究各自的人工智能(artificial intelligence,ai)芯片以满足在ai领域的差异化竞争,从而获得领先地位。业界目前主要基于张量虚拟机(tensor virtualmachine,tvm)等ai编译框架在特定硬件芯片上对关键算子进行加速,实现高性能神经网络模型的搭建。

2、上述的两种ai编译框架,在开发者所使用的高级语言中是基于计算描述(如计算compute)与调度策略(如调优schedule)分离的概念来描述算子逻辑。其中,compute专注于计算逻辑的描述,schedule提供指令以及抽象语法树(abstract syntax tree,ast)变化的能力,并利用框架提供的原语能力对compute进行优化。

<p>3、排序算子是现本文档来自技高网...

【技术保护点】

1.一种排序算子的编译方法,其特征在于,应用于编译器,所述方法包括:

2.根据权利要求1所述的方法,其特征在于,

3.根据权利要求1或2所述的方法,其特征在于,

4.根据权利要求3所述的方法,其特征在于,

5.根据权利要求4所述的方法,其特征在于,

6.根据权利要求4或5所述的方法,其特征在于,

7.根据权利要求1-6中任一项所述的方法,其特征在于,

8.一种编译装置,其特征在于,所述装置包括:

9.根据权利要求8所述的装置,其特征在于,

10.根据权利要求8或9所述的装置,其特征...

【技术特征摘要】

1.一种排序算子的编译方法,其特征在于,应用于编译器,所述方法包括:

2.根据权利要求1所述的方法,其特征在于,

3.根据权利要求1或2所述的方法,其特征在于,

4.根据权利要求3所述的方法,其特征在于,

5.根据权利要求4所述的方法,其特征在于,

6.根据权利要求4或5所述的方法,其特征在于,

7.根据权利要求1-6中任一项所述的方法,其特征在于,

8.一种编译装置,其特征在于,所述装置包括:

9.根据权利要求8所述的装置,其特征在于,

10.根据权利要求8或9所述的装置,其特征在于,

11.根据权利要求10所述的装置,其特征在于,

12.根据权利要求11所述的装置,其特征在于,

13.根据权利要求11或12所述的装置,其特征在于,

14.根据权利要求8-12中任一项所述...

【专利技术属性】
技术研发人员:金格程琛吴凌飞徐子明卢建鹏尹科杰
申请(专利权)人:华为技术有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1