【技术实现步骤摘要】
用于神经网络计算的应用编程接口
[0001]至少一个实施例涉及用于改进神经网络计算的处理资源。例如,至少一个实施例涉及根据本文所描述的各个新颖技术,用于使用由一个或更多个用户提供的描述符来提供库以促进优化的神经网络操作的处理器或计算系统。
技术介绍
[0002]使用并行处理单元(如图形处理单元)来加速深度学习已经导致基于软件的库来提供各个深度学习操作的实现。这些基于软件的深度学习库提供了由软件程序用来执行深度学习操作(诸如神经网络计算)的应用编程接口。然而,可用的深度学习库常常无法为各个并行处理单元可用的所有优化提供支持,并且可能具有必须随着对所述深度学习库增加功能而改变的笨重的应用编程接口。这导致使用这些深度学习库的软件程序的可维护性和性能降低。
附图说明
[0003]图1是示出了根据至少一个实施例的用于由一个或更多个神经网络优化计算的深度神经网络库架构的框图;
[0004]图2是示出了根据至少一个实施例的用于根据由深度神经网络库接收的一个或更多个描述符生成优化操作集的架构的框图;
[0005]图3是示出了根据至少一个实施例的用于使用深度神经网络库来选择一个或更多个计算选项的架构的框图;
[0006]图4是示出了根据至少一个实施例的用于使用深度神经网络库确定在执行计划中使用的中间体的架构的框图;
[0007]图5是示出了根据至少一个实施例的用于使用深度神经网络库来管理数据指针的架构的框图;
[0008]图6是示出了根据至少一个实施例的用于使用深度神经网络库结合 ...
【技术保护点】
【技术特征摘要】
1.一种机器可读介质,其具有存储在其上的应用编程接口API调用,所述API调用如果由一个或更多个处理器执行,则使所述一个或更多个处理器至少:使由第二API调用指示的一个或更多个操作被执行。2.根据权利要求1所述的机器可读介质,还包括指令,所述指令如果由所述一个或更多个处理器执行,则使所述一个或更多个处理器:作为所述第二API调用的结果,接收与所述一个或更多个操作相关联的一个或更多个描述符。3.根据权利要求2所述的机器可读介质,其中所述一个或更多个描述符指示由所述一个或更多个操作可用的一个或更多个数据值以及将由所述一个或更多个操作执行的一个或更多个计算。4.根据权利要求2所述的机器可读介质,其中所述指令还包括以下指令:作为由所述一个或更多个处理器执行的结果,该指令使所述一个或更多个处理器:识别将被应用于所述一个或更多个描述符的一个或更多个优化;以及作为第三API调用的结果,至少部分地基于将所述一个或更多个优化应用于所述一个或更多个描述符而生成所述一个或更多个操作。5.根据权利要求1所述的机器可读介质,还包括以下指令:该指令如果由所述一个或更多个处理器执行,则使所述一个或更多个处理器:使用所述API传输第一组计算选项;以及作为第三API调用的结果,接收能用于配置所述一个或更多个操作的第二组计算选项,所述第二组计算选项是所述第一组计算选项的子集。6.根据权利要求1所述的机器可读介质,还包括以下指令:该指令如果由所述一个或更多个处理器执行,则使所述一个或更多个处理器:作为第三API调用的结果,接收要与所述一个或更多个操作结合使用的一个或更多个预计算的数据值;以及将所述一个或更多个预计算的数据值与所述一个或更多个操作相关联。7.根据权利要求1所述的机器可读介质,还包括以下指令:该指令如果由所述一个或更多个处理器执行,则使所述一个或更多个处理器:作为第三API调用的结果,接收要与所述一个或更多个操作结合使用的指向用户分配的存储器的一组指针;以及将指向用户分配的存储器的该组指针与所述一个或更多个操作相关联。8.根据权利要求1所述的机器可读介质,其中所述API与深度神经网络库相关联,并且所述一个或更多个操作是作为所述API调用的结果而由所述深度神经网络库执行的神经网络操作。9.一种方法,包括:作为对应用编程接口API的第一函数调用的结果,执行由对所述API的第二函数调用指示的一个或更多个操作。10.根据权利要求9所述的方法,还包括:至少部分地基于结合所述第二函数调用接收的一个或更多个描述符来确定所述一个或更多个操作。11.根据权利要求9所述的方法,还包括:
至少部分地基于所述一个或更多个操作是有效的确定来生成指示;以及使用所述API传输所述指示。12.根据权利要求9所述的方法,还包括:至少部分地基于将一个或更多个性能优化应用于所述一个或更多个操作来生成操作集,所述一个或更多个性能优化由对所述API的第三函数调用指示;以及作为所述第一函数调用的结果,执行所述操作。13.根据权利要求9所述的方法,还包括:作为对所述API的第三函数调用的结果,传输与所述一个或更多个操作相关联的第一组计算选项;作为对所述API的第四函数调用的结果,接收第二组计算选项,所述第二组是所述第一组的子集;以及将所述第二组计算选项与所述一个或更多个操作相关联。14.根据权利要求9所述的方法,还包括:确定与所述一个或更多个操作相关联的一个或更多个数据依赖关系;使用所述API传输所述一个或更多个数据依赖关系;以及与所述一个或更多个操作相关联地存储一个或更多个预计算的数据值,作为传输所述一个或更多个数据依赖关系的结果,所述一个或更多个预计算的数据值经由对所述API的第三函数调用而接收。15.根据权利要求9所述的方法,还包括:作为对所述AP...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。