CN118674842A - 使用低分辨率光流的硬件高效神经帧预测 - Google Patents

使用低分辨率光流的硬件高效神经帧预测 Download PDF

Info

Publication number
CN118674842A
CN118674842A CN202311697886.5A CN202311697886A CN118674842A CN 118674842 A CN118674842 A CN 118674842A CN 202311697886 A CN202311697886 A CN 202311697886A CN 118674842 A CN118674842 A CN 118674842A
Authority
CN
China
Prior art keywords
data
optical flow
graphics
memory
resolution
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Pending
Application number
CN202311697886.5A
Other languages
English (en)
Inventor
D·R·艾耶
D·范巴尔
王长亮
S·巴蒂亚
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Intel Corp
Original Assignee
Intel Corp
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Priority claimed from US18/308,725 external-priority patent/US20240311962A1/en
Application filed by Intel Corp filed Critical Intel Corp
Publication of CN118674842A publication Critical patent/CN118674842A/zh
Pending legal-status Critical Current

Links

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T15/003D [Three Dimensional] image rendering
    • G06T15/005General purpose rendering architectures
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/044Recurrent networks, e.g. Hopfield networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • G06N3/0455Auto-encoder networks; Encoder-decoder networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/0464Convolutional networks [CNN, ConvNet]
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • G06N3/084Backpropagation, e.g. using gradient descent
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T1/00General purpose image data processing
    • G06T1/20Processor architectures; Processor configuration, e.g. pipelining
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T1/00General purpose image data processing
    • G06T1/60Memory management
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T3/00Geometric image transformations in the plane of the image
    • G06T3/40Scaling of whole images or parts thereof, e.g. expanding or contracting
    • G06T3/4046Scaling of whole images or parts thereof, e.g. expanding or contracting using neural networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T3/00Geometric image transformations in the plane of the image
    • G06T3/40Scaling of whole images or parts thereof, e.g. expanding or contracting
    • G06T3/4053Scaling of whole images or parts thereof, e.g. expanding or contracting based on super-resolution, i.e. the output image resolution being higher than the sensor resolution
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T5/00Image enhancement or restoration
    • G06T5/20Image enhancement or restoration using local operators
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/20Analysis of motion
    • G06T7/269Analysis of motion using gradient-based methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2200/00Indexing scheme for image data processing or generation, in general
    • G06T2200/28Indexing scheme for image data processing or generation, in general involving image processing hardware
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20081Training; Learning
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20084Artificial neural networks [ANN]

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Evolutionary Computation (AREA)
  • Artificial Intelligence (AREA)
  • General Health & Medical Sciences (AREA)
  • General Engineering & Computer Science (AREA)
  • Data Mining & Analysis (AREA)
  • Biophysics (AREA)
  • Biomedical Technology (AREA)
  • Molecular Biology (AREA)
  • Computing Systems (AREA)
  • Computational Linguistics (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Mathematical Physics (AREA)
  • Software Systems (AREA)
  • Health & Medical Sciences (AREA)
  • Multimedia (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Computer Graphics (AREA)
  • Image Generation (AREA)

Abstract

本文中描述了用于使用经上采样的光流数据、经由神经帧生成针对3D渲染应用增强用户体验的技术。在一个实施例中,使用稀疏光流数据和密集光流数据两者来训练神经网络以使得神经帧生成能够由部署的神经网络仅使用稀疏光流数据来执行。稀疏光流数据可以由经训练的神经网络上采样到密集光流数据。神经网络可以使用经上采样的密集光流数据来执行帧生成。

Description

使用低分辨率光流的硬件高效神经帧预测
交叉引用
本专利申请要求于2023年3月16日提交的临时申请第63/490,609号的优先权,该申请的内容通过引用以其整体并入本文中。
技术领域
本公开总体上涉及数据处理,并且更具体地,涉及经由内插和外推两者的神经帧生成。
背景技术
针对渲染应用使用神经技术来生成帧在计算领域中是感兴趣的。这些技术可以分成两类:内插技术,其中帧在两个渲染帧之间被生成;以及外推技术,其中未来帧基于一些过去的输入帧被预测。每种方法在图像质量、等待时间、网络体系结构和性能方面都有其优缺点。针对这些用例中的每个用例开发神经网络都是详尽的过程,该过程要求多次迭代以在质量与性能之间实现正确的权衡。
附图说明
在所附附图的各图中以示例方式而非限制方式来图示本文中描述的实施例,在附图中,类似的附图标记指示类似的要素,并且其中:
图1是图示配置成用于实现本文中描述的实施例的一个或多个方面的计算机系统的框图;
图2A-图2D图示并行处理器部件;
图3A-图3C是图形多处理器和基于多处理器的GPU的框图;
图4A-图4F图示在其中多个GPU通信地耦合至多个多核心处理器的示例性体系结构;
图5图示图形处理管线;
图6图示机器学习软件栈;
图7图示通用图形处理单元;
图8图示多GPU计算系统;
图9A-图9B图示示例性深度神经网络的层;
图10图示示例性循环神经网络;
图11图示深度神经网络的训练和部署;
图12A是图示分布式学习的框图;
图12B是图示可编程网络接口和数据处理单元的框图;
图13图示适于使用经训练的模型执行推断的示例性推断片上系统(system on achip,SOC);
图14是处理系统的框图;
图15A-图15C图示计算系统和图形处理器;
图16A-图16C图示附加的图形处理器和计算加速器体系结构的框图;
图17是图形处理器的图形处理引擎的框图;
图18A-图18C图示包括在图形处理器核心中采用的处理元件的阵列的线程执行逻辑;
图19图示根据实施例的多片处理器的片;
图20是图示图形处理器指令格式的框图;
图21是附加的图形处理器体系结构的框图;
图22A-图22B图示图形处理器命令格式和命令序列;
图23图示用于数据处理系统的示例性图形软件体系结构;
图24A是图示IP核心开发系统的框图;
图24B图示集成电路封装组件的截面侧视图;
图24C图示封装组件,该封装组件包括连接到衬底(例如,基础管芯)的多个单元的硬件逻辑小芯片;
图24D图示包括可互换小芯片的封装组件;
图25是图示示例性片上系统集成电路的框图;
图26A-图26B是图示用于在SoC内使用的示例性图形处理器的框图;
图27是根据实施例的数据处理系统的框图;
图28A-图28B图示根据实施例的由指令管线执行的矩阵操作;
图29图示计算块,该计算块包括启用编解码器的分解的脉动逻辑;
图30A-图30B图示根据实施例的、用于使用统一机器学习模型进行帧外推和帧内插的系统。
图31A-图31B图示根据实施例的、用于使用统一机器学习模型进行帧外推和帧内插的方法。
图32图示根据实施例的、用于帧外推和帧内插的神经网络。
图33图示根据实施例的去噪网络;
图34A-图34B图示根据实施例的、用于训练和操作统一内插/外推网络的方法。
图35A-图35B图示用于使用低分辨率光流实现推断的训练操作;
图36图示根据实施例的包括光流上采样的帧生成;
图37图示根据实施例的光流上采样和去噪网络;
图38A-图38B图示根据实施例的、训练和操作具有光流上采样和去噪子网络以及光流估计子网络的神经网络的方法;
图39是根据实施例的包括图形处理器的计算设备的框图。
具体实施方式
当前的并行图形数据处理包括被开发成对图形数据执行特定操作的系统和方法,这些特定操作诸如例如,线性内插、曲面细分、栅格化、纹理映射、深度测试等。传统意义上而言,图形处理器使用固定功能计算单元来处理图形数据。然而,更最近地,已使图形处理器的多个部分可编程,使得此类处理器能够支持更广泛种类的操作以处理顶点数据和片段数据。
为了进一步提升性能,图形处理器典型地实现诸如管线化之类的处理技术,这些处理技术尝试贯穿图形管线的不同部分并行地处理尽可能多的图形数据。具有单指令多线程(single instruction,multiple thread,SIMT)体系结构的并行图形处理器被设计成使图形管线中的并行处理的量最大化。在SIMT体系结构中,成组的并行线程尝试尽可能频繁地一起同步地执行程序指令以提高处理效率。可在Shane Cook的“CUDA编程”第3章第37-51页(2013年)中找到用于SIMT体系结构的软件和硬件的总体概述。
图形处理单元(graphics processing unit,GPU)通信地耦合至主机/处理器核心以加速例如图形操作、机器学习操作、模式分析操作、和/或各种通用GPU(general-purposeGPU,GPGPU)功能。GPU可通过总线或另一互连(例如,诸如PCIe或NVLink之类的高速互连)通信地耦合至主机处理器/核心。替代地,GPU可集成在与核心相同的封装或芯片上,并且通过内部处理器总线/互连(即,在封装或芯片内部)通信地耦合至核心。无论GPU被连接所采取的方式如何,处理器核心都可将工作以工作描述符中所包含的命令/指令序列的形式分配给GPU。GPU随后使用专用电路/逻辑来高效地处理这些命令/指令。
在以下描述中,陈述了众多特定细节以提供更透彻的理解。然而,对于本领域的技术人员将显而易见的是,可以在没有这些特定细节中的一个或多个特定细节的情况下实践本文中描述的实施例。在其他实例中,未描述公知的特征以免混淆当前实施例的细节。
在各实施例中,本文中描述了用于经由神经帧生成来增强针对3D渲染应用的用户体验的技术。一个实施例提供了一种技术,该技术用于使用稀疏光流数据和密集光流数据两者来训练神经网络以使得神经帧生成能够由部署的神经网络仅使用稀疏光流数据来执行。稀疏光流数据可以在训练期间并且由经训练的神经网络上采样到密集光流数据。可以训练神经网络以将稀疏光流数据准确地上采样到密集光流数据。然后,密集光流数据可用于执行帧生成。
附加地,一个实施例提供了用于帧内插和外推的等待时间感知统一神经网络。这种统一神经网络将内插和外推网络合并成一个广义网络,该广义网络可被应用于内插和外推两者,这取决于可接受的性能的等待时间。尽管用于两者的网络体系结构相同,但仅权重基于目标应用和期望的等待时间度量而改变,从而使网络体系结构适用于内插和外推两者,而无需重新训练和重新开发新网络。
系统概览
图1是图示配置成用于实现本文中描述的实施例的一个或多个方面的计算系统100的框图。计算系统100包括处理子系统101,该处理子系统101具有经由互连路径通信的一个或多个处理器102和系统存储器104,该互连路径可包括存储器中枢105。存储器中枢105可以是芯片组部件内的单独部件,或者可被集成在一个或多个处理器102内。存储器中枢105经由通信链路106与I/O子系统111耦合。I/O子系统111包括I/O中枢107,该I/O中枢107可使计算系统100能够从一个或多个输入设备108接收输入。此外,I/O中枢107可使显示控制器(其可被包括在一个或多个处理器102中)将输出提供给一个或多个显示设备110A。在一个实施例中,与I/O中枢107耦合的一个或多个显示设备110A可包括本地的、内部的、或嵌入式的显示设备。
处理子系统101例如包括经由总线或其他通信链路113耦合至存储器中枢105的一个或多个并行处理器112。通信链路113可以是任何数量的基于标准的通信链路技术或协议中的一种,诸如但不限于PCI快速(PCI Express),或者可以是供应方特定的通信接口或通信结构(fabric)。一个或多个并行处理器112可形成可包括大量处理核心和/或处理集群的计算集中的并行或向量处理系统,诸如,集成众核心(many integrated core,MIC)处理器。例如,一个或多个并行处理器112形成图形处理子系统,该图形处理子系统可以向经由I/O中枢107耦合的一个或多个显示设备110A中的一个显示设备输出像素。一个或多个并行处理器112还可包括显示控制器和显示接口(未示出),用于启用至一个或多个显示设备110B的直接连接。
在I/O子系统111内,系统存储单元114可连接到I/O中枢107,从而为计算系统100提供存储机制。I/O开关116可用于提供接口机制,以启用I/O中枢107与其他部件之间的连接,其他部件诸如,可被集成到平台中的网络适配器118和/或无线网络适配器119、以及可经由一个或多个插入式设备120而被添加的各种其他设备。(一个或多个)插入式设备120还可包括例如一个或多个外部图形处理器设备、图形卡、和/或计算加速器。网络适配器118可以是以太网适配器或另一有线网络适配器。无线网络适配器119可包括以下一者或多者:Wi-Fi、蓝牙、近场通信(near field communication,NFC)、或包括一个或多个无线的无线电装置的其他网络设备。
计算系统100可包括未显式地示出的其他部件,包括USB或其他端口连接、光学存储驱动器、视频捕捉设备等等,这些部件也可连接到I/O中枢107。将图1中的各种部件互连的通信路径可使用任何合适的协议来实现,合适的协议诸如,基于PCI(PeripheralComponent Interconnect,外围部件互连)的协议(例如,PCI快速)、或任何其他总线或点到点通信接口和/或(一种或多种)协议,诸如,NVLink高速互连、计算快速链路TM(ComputeExpress LinkTM,CXLTM)(例如,CXL.mem)、无限结构(Infinity Fabric,IF)、以太网(IEEE802.3)、远程直接存储器访问(remote direct memory access,RDMA)、无限带宽(InfiniBand)、网际广域RDMA协议(Internet Wide Area RDMA Protocol,iWARP)、传输控制协议(Transmission Control Protocol,TCP)、用户数据报协议(User DatagramProtocol,UDP)、快速UDP网际连接(quick UDP Internet Connections,QUIC)、通过汇聚以太网的RDMA(RDMA over Converged Ethernet,RoCE)、英特尔快速路径互连(IntelQuickPath Interconnect,QPI)、英特尔超路径互连(Intel Ultra Path Interconnect,UPI)、英特尔片上系统结构(Intel On-Chip System Fabric,IOSF)、全方位路径(Omnipath)、超传输(HyperTransport)、高级微控制器总线体系结构(AdvancedMicrocontroller Bus Architecture,AMBA)互连、OpenCAPI、Gen-Z、用于加速器的缓存一致互连(Cache Coherent Interconnect for Accelerators,CCIX)、3GPP长期演进(3GPPLong Term Evolution,LTE)(4G)、3GPP 5G及其变体、或本领域中已知的有线或无线互连协议。在一些示例中,可使用诸如通过结构的非易失性存储器快速(non-volatile memoryexpress,NVMe)(non-volatile memory express over Fabrics,NVME-oF)或NVMe之类的协议将数据复制或存储到虚拟化存储节点。
一个或多个并行处理器112可包含针对图形和视频处理进行优化的电路(包括例如,视频输出电路),并构成图形处理单元(GPU)。替代地或附加地,本文中更详细地描述,一个或多个并行处理器112可包含针对通用处理进行优化同时保留底层计算体系结构的电路。计算系统100的部件可与一个或多个其他系统元件集成在单个集成电路上。例如,一个或多个并行处理器112、存储器中枢105、(一个或多个)处理器102、以及I/O中枢107可被集成到片上系统(SoC)集成电路中。替代地,计算系统100的部件可被集成到单个封装中以形成系统级封装(systemin package,SIP)配置。在一个实施例中,计算系统100的部件的至少部分可被集成到多芯片模块(multi-chip module,MCM)中,该MCM可与其他多芯片模块一起被互连到模块化计算系统中。
将领会,本文中所示出的计算系统100是说明性的,并且变体和修改是可能的。可根据需要修改连接拓扑,包括桥接器的数量和布置、(一个或多个)处理器102的数量、以及(一个或多个)并行处理器112的数量。例如,系统存储器104可直接而不是通过桥接器连接到(一个或多个)处理器102,而其他设备经由存储器中枢105和(一个或多个)处理器102与系统存储器104通信。在其他替代拓扑中,(一个或多个)并行处理器连接到I/O中枢107或直接连接到一个或多个处理器102中的一个处理器,而不是连接到存储器中枢105。在其他实施例中,I/O中枢107和存储器中枢105可集成到单个芯片中。两个或更多个处理器102集合经由多个插槽被附连也是可能的,多个插槽可与(一个或多个)并行处理器112的两个或更多个实例耦合。
本文中所示出的特定部件中的一些是任选的,并且可以是并非在计算系统100的所有实现方式中都包括这些部件。例如,可支持任何数量的插入式卡或外围设备,或者可消除一些部件。此外,一些体系结构可针对与图1中所图示的那些部件类似的部件使用不同的术语。例如,存储器中枢105在一些体系结构中可被称为北桥,而I/O中枢107可被称为南桥。
图2A图示并行处理器200。并行处理器200可以是如本文中所述的GPU、GPGPU等。并行处理器200的各种部件可使用一个或多个集成电路设备来实现,一个或多个集成电路设备诸如,可编程处理器、专用集成电路(application specific integrated circuit,ASIC)、或现场可编程门阵列(field programmable gate array,FPGA)。所图示的并行处理器200可以是图1中示出的(一个或多个)并行处理器112中的一个或多个。
并行处理器200包括并行处理单元202。并行处理单元包括启用与其他设备的通信的I/O单元204,其他设备包括并行处理单元202的其他实例。I/O单元204可直接连接到其他设备。例如,I/O单元204经由使用中枢或开关接口(诸如,存储器中枢105)与其他设备连接。存储器中枢105与I/O单元204之间的连接形成通信链路113。在并行处理单元202内,I/O单元204与主机接口206以及存储器交叉开关216连接,其中,主机接口206接收涉及执行处理操作的命令,并且存储器交叉开关216接收涉及执行存储器操作的命令。
当主机接口206经由I/O单元204接收命令缓冲器时,主机接口206可以将用于执行那些命令的工作操作引导至前端208。在一个实施例中,前端208与调度器210耦合,该调度器210被配置成用于将命令或其他工作项目分发给处理集群阵列212。调度器210确保在任务被分发给处理集群阵列212中的处理集群之前,处理集群阵列212被适当地配置并且处于有效状态。调度器210可经由在微控制器上执行的固件逻辑来实现。微控制器实现的调度器210可被配置成用于以粗粒度和细粒度执行复杂的调度和工作分发操作,从而实现对在处理集群阵列212上执行的线程的快速抢占和上下文切换。优选地,主机软件可以经由多个图形处理门铃中的一个来证实用于在处理集群阵列212上调度的工作负载。在其他示例中,可使用对新工作负载或中断的轮询来标识或指示要执行的工作的可用性。工作负载随后可由调度器微控制器内的调度器210逻辑跨处理集群阵列212自动地分发。
处理集群阵列212可以包括最多“N”个处理集群(例如,集群214A、集群214B至集群214N)。处理集群阵列212中的每个集群214A-214N可执行大量的并发线程。调度器210可以使用各种调度和/或工作分发算法将工作分配给处理集群阵列212中的集群214A-214N,这些调度和/或工作分发算法可取决于针对每种类型的程序或计算产生的工作负载而变化。调度可以由调度器210动态地处置,或可以在对被配置成供处理集群阵列212执行的程序逻辑的编译期间部分地由编译器逻辑辅助。任选地,处理集群阵列212中的不同的集群214A-214N可以被分配用于处理不同类型的程序或用于执行不同类型的计算。
处理集群阵列212可被配置成用于执行各种类型的并行处理操作。例如,处理集群阵列212被配置成用于执行通用并行计算操作。例如,处理集群阵列212可包括用于执行处理任务的逻辑,这些处理任务包括视频和/或音频数据的过滤、执行包括物理操作的建模操作以及执行数据变换。
处理集群阵列212被配置成用于执行并行图形处理操作。在其中并行处理器200被配置成用于执行图形处理操作的此类实施例中,处理集群阵列212可包括用于支持此类图形处理操作的执行的附加逻辑,包括但不限于用于执行纹理操作的纹理采样逻辑、以及曲面细分逻辑和其他顶点处理逻辑。此外,处理集群阵列212可被配置成用于执行图形处理相关的着色器程序,诸如但不限于顶点着色器、曲面细分着色器、几何着色器和像素着色器。并行处理单元202可以经由I/O单元204从系统存储器传输数据以供处理。在处理期间,可在处理期间将所传输的数据存储到片上存储器(例如,并行处理器存储器222),随后将该数据写回到系统存储器。
在其中使用并行处理单元202来执行图形处理的实施例中,调度器210可被配置成用于将处理工作负载划分成近似相等大小的任务,以更好地实现图形处理操作向处理集群阵列212中的多个集群214A-214N的分发。在这些实施例中的一些实施例中,处理集群阵列212的部分可被配置成用于执行不同类型的处理。例如,第一部分可被配置成用于执行顶点着色和拓扑生成,第二部分可被配置成用于执行曲面细分和几何着色,并且第三部分可被配置成用于执行像素着色或其他屏幕空间操作,以产生用于显示的经渲染的图像。由集群214A-214N中的一个或多个集群产生的中间数据可被存储在缓冲器中,以允许中间数据在集群214A-214N之间被传送以用于进一步处理。
在操作期间,处理集群阵列212可以经由调度器210接收要被执行的处理任务,该调度器210从前端208接收定义处理任务的命令。对于图形处理操作,处理任务可包括要被处理的数据以及定义将如何处理该数据(例如,将执行什么程序)的状态参数和命令的索引,该数据例如,表面(补片(patch))数据、基元数据、顶点数据和/或像素数据。调度器210可被配置成用于取得(fetch)与任务对应的索引,或者可从前端208接收索引。前端208可被配置成用于确保在由传入命令缓冲器(例如,批量缓冲器、推入缓冲器等)指定的工作负载被发起之前处理集群阵列212被配置成有效状态。
并行处理单元202的一个或多个实例中的每个实例可以与并行处理器存储器222耦合。可以经由存储器交叉开关216来访问并行处理器存储器222,该存储器交叉开关216可以接收来自处理集群阵列212以及I/O单元204的存储器请求。存储器交叉开关216可经由存储器接口218来访问并行处理器存储器222。存储器接口218可包括各自可耦合至并行处理器存储器222的部分(例如,存储器单元)的多个分区单元(例如,分区单元220A、分区单元220B、直到分区单元220N)。分区单元220A-220N的数量可被配置成等于存储器单元的数量,使得第一分区单元220A具有对应的第一存储器单元224A,第二分区单元220B具有对应的第二存储器单元224B,并且第N分区单元220N具有对应的第N存储器单元224N。在其他实施例中,分区单元220A-220N的数量可以不等于存储器设备的数量。
存储器单元224A-224N可以包括各种类型的存储器设备,包括动态随机存取存储器(dynamic random-access memory,DRAM)或图形随机存取存储器,诸如,同步图形随机存取存储器(synchronous graphics random access memory,SGRAM),包括图形双倍数据速率(graphics double data rate,GDDR)存储器。任选地,存储器单元224A-224N还可包括3D堆叠式存储器,包括但不限于高带宽存储器(high bandwidth memory,HBM)。本领域技术人员将领会,存储器单元224A-224N的具体实现方式可有所不同,并且可从各种常规设计中的一个常规设计中选择。诸如帧缓冲器或纹理图之类的渲染目标可跨存储器单元224A-224N被存储,从而允许分区单元220A-220N并行地写入每个渲染目标的部分,以高效地使用并行处理器存储器222的可用带宽。在一些实施例中,并行处理器存储器222的本地实例可被排除,以有利于利用结合本地缓存存储器的系统存储器的统一存储器设计。
任选地,处理集群阵列212中的集群214A-214N中的任一者具有用于处理将被写入到并行处理器存储器222内的存储器单元224A-224N中的任一者的数据的能力。存储器交叉开关216可被配置成用于将每个集群214A-214N的输出传输到任何分区单元220A-220N或传输到另一集群214A-214N,该另一集群214A-214N可对输出执行附加的处理操作。每个集群214A-214N可通过存储器交叉开关216与存储器接口218通信,以从各种外部存储器设备读取或向各种外部存储器设备写入。在具有存储器交叉开关216的实施例中的一个实施例中,存储器交叉开关216具有至存储器接口218的连接以与I/O单元204通信,并具有至并行处理器存储器222的本地实例的连接,从而使不同处理集群214A-214N内的处理单元能够与系统存储器或不在并行处理单元202本地的其他存储器通信。一般而言,存储器交叉开关216例如可以能够使用虚拟通道来分离集群214A-214N与分区单元220A-220N之间的通信量流。
尽管在并行处理器200内图示出并行处理单元202的单个实例,但可以包括并行处理单元202的任何数量的实例。例如,并行处理单元202的多个实例可以被设置在单个插入式卡上,或者多个插入式卡可以被互连。例如,并行处理器200可以是插入式设备,诸如,图1的插入式设备120,该插入式设备120可以是图形卡(诸如,包括一个或多个GPU、一个或多个存储器设备、以及设备至设备或网络或结构接口的分立的图形卡)。并行处理单元202的不同实例可被配置成即便不同实例具有不同数量的处理核心、不同量的本地并行处理器存储器、和/或其他配置区别也进行互操作。任选地,并行处理单元202的一些实例相对于其他实例可包括更高精度浮点单元。包含并行处理单元202或并行处理器200的一个或多个实例的系统能以各种配置和形状因子来实现,这些配置和形状因子包括但不限于,桌面型电脑、膝上型电脑、或手持式个人计算机、服务器、工作站、游戏控制台和/或嵌入式系统。编排器可使用以下一者或多者来形成用于工作负载执行的复合节点:分解的处理器资源、缓存资源、存储器资源、存储资源、和联网资源。
在一个实施例中,并行处理单元202可以被分区为多个实例。那些多个实例可以被配置用于以隔离的方式执行与不同客户端相关联的工作负载,从而使得为每个客户端提供预定的服务质量。例如,每个集群214A-214N可以与其他集群划分开并隔离开,从而允许处理集群阵列212被划分为多个计算分区或实例。在此类配置中,在隔离分区上执行的工作负载被保护以免受与在不同分区上执行的不同工作负载相关联的错误或误差。分区单元220A-220N可以被配置成用于启用到与相应的计算分区相关联的集群214A-214N的存储器的专用路径和/或隔离路径。此数据路径隔离使得分区内的计算资源可与一个或多个经指派的存储器单元224A-224N进行通信,而不会受到其他分区的活动的干扰。
图2B是分区单元220的框图。分区单元220可以是图2A的分区单元220A-220N中的一个分区单元的实例。如所图示,分区单元220包括L2缓存221、帧缓冲器接口225、和ROP226(栅格操作单元)。L2缓存221是被配置成用于执行从存储器交叉开关216和ROP 226接收的加载和存储操作的读取/写入缓存。读取未命中和紧急写回请求由L2缓存221输出到帧缓冲器接口225以用于处理。更新也可经由帧缓冲器接口225被发送到帧缓冲器以用于处理。在一个实施例中,帧缓冲器接口225对图2A的并行处理器存储器222内的存储器单元224A-224N中的存储器单元224提供接口。分区单元220还可附加地或替代地经由存储器控制器(未示出)与并行处理器存储器中的存储器单元中的一个存储器单元对接。
在图形应用中,ROP 226是执行栅格操作(诸如,模板印制(stencil)、z测试、混合等等)的处理单元。ROP 226随后输出经处理的图形数据,该经处理的图形数据被存储在图形存储器中。在一些实施例中,ROP 226包括编解码器(CODEC)227或与CODEC 227耦合,该CODEC 227包括压缩逻辑,该压缩逻辑用于对被写入存储器或L2缓存221的深度或颜色数据进行压缩,并对从存储器或L2缓存221读取的深度或颜色数据进行解压缩。压缩逻辑可以是利用多种压缩算法中的一种或多种的无损压缩逻辑。由CODEC 227执行的压缩的类型可以基于将要被压缩的数据的统计特性而变化。例如,在一个实施例中,逐片地对深度和颜色数据执行Δ(delta)颜色压缩。在一个实施例中,CODEC 227包括压缩和解压缩逻辑,该压缩和解压缩逻辑可对与机器学习操作相关联的计算数据进行压缩和解压缩。CODEC 227可例如对用于稀疏机器学习操作的稀疏矩阵数据进行压缩。CODEC 227还可对按稀疏矩阵格式(例如,坐标列表编码(coordinate list encoding,COO)、压缩稀疏行(compressed sparserow,CSR)、压缩稀疏列(compress sparse column,CSC)等)编码的稀疏矩阵数据进行压缩以生成经压缩且经编码的稀疏矩阵数据。经压缩且经编码的稀疏矩阵数据可在由处理元件处理之前被解压缩和/或解码,或者处理元件可被配置成用于消耗经压缩的、经编码的、或经压缩且经编码的数据以用于处理。
ROP 226可被包括在每个处理集群(例如,图2A的集群214A-214N)内而不是被包括在分区单元220内。在此类实施例中,通过存储器交叉开关216来传送对像素数据而非像素片段数据的读取和写入请求。经处理的图形数据可被显示在显示设备(诸如,图1的一个或多个显示设备110A-110B中的一个显示设备)上,被路由以用于由(一个或多个)处理器102进一步处理,或者被路由以用于由图2A的并行处理器200内的处理实体中的一个处理实体进一步处理。
图2C是并行处理单元内的处理集群214的框图。例如,处理集群是图2A的处理集群214A-214N中的一个处理集群的实例。处理集群214可被配置成用于并行地执行许多线程,其中,术语“线程”是指对特定的输入数据的集合执行的特定程序的实例。任选地,可使用单指令多数据(single-instruction,multiple-data,SIMD)指令发出技术以在不提供多个独立的指令单元的情况下支持大量线程的并行执行。替代地,可使用单指令多线程(single-instruction,multiple-thread,SIMT)技术来使用被配置成用于向处理集群中的每个处理集群内的处理引擎的集合发出指令的共同的指令单元来支持大量总体上同步的线程的并行执行。与其中所有处理引擎典型地执行相同指令的SIMD执行机制不同,SIMT执行允许不同的线程更容易地遵循通过给定的线程程序的发散的执行路径。本领域技术人员将理解,SIMD处理机制表示SIMT处理机制的功能子集。
可以经由将处理任务分发给SIMT并行处理器的管线管理器232来控制处理集群214的操作。管线管理器232接收来自图2A的调度器210的指令,并且经由图形多处理器234和/或纹理单元236来管理那些指令的执行。所图示的图形多处理器234是SIMT并行处理器的示例性实例。然而,可将不同体系结构的各种类型的SIMT并行处理器包括在处理集群214内。图形多处理器234的一个或多个实例可被包括在处理集群214内。图形多处理器234可以处理数据,并且数据交叉开关240可以用于将经处理的数据分发到多个可能的目的地中的一个目的地,包括促进处理集群214内的图形多处理器之间的数据的交换。管线管理器232可通过指定用于要经由数据交叉开关240分发的经处理的数据的目的地来促进经处理的数据的分发。
处理集群214内的每个图形多处理器234可以包括相同的功能执行逻辑集合(例如,算术逻辑单元、加载-存储单元等)。能以管线化的方式配置功能执行逻辑,按照该管线化的方式,新指令可在先前指令完成之前被发出。功能执行逻辑支持各种操作,包括整数和浮点算术、比较操作、布尔操作、比特移位、以及各种代数函数的计算。可利用相同的功能单元硬件来执行不同的操作,并且功能单元的任何组合可以存在。
被传送至处理集群214的指令构成线程。跨并行处理引擎的集合执行的线程的集合是线程组。线程组对不同的输入数据执行相同的程序。线程组内的每个线程可被指派给图形多处理器234内的不同的处理引擎。线程组可包括比图形多处理器234内的处理引擎的数量更少的线程。当线程组包括比处理引擎的数量更少的线程时,处理引擎中的一个或多个处理引擎在其间线程组正被处理的周期期间可以是空闲的。线程组也可包括比图形多处理器234内的处理引擎的数量更多的线程。当线程组包括比图形多处理器234内的处理引擎的数量更多的线程时,可在连续的时钟周期内执行处理。任选地,可在图形多处理器234上并发地执行多个线程组。
图形多处理器234可包括内部缓存存储器,以执行加载和存储操作。任选地,图形多处理器234可放弃内部缓存,并使用处理集群214内的缓存存储器(例如,第一级(level1,L1)缓存248)。每个图形多处理器234还具有对分区单元(例如,图2A的分区单元220A-220N)内的第二级(L2)缓存的访问权,这些L2缓存在所有处理集群214之间被共享,并且可被用于在线程之间传输数据。图形多处理器234还可访问片外全局存储器,该片外全局存储器可包括本地并行处理器存储器和/或系统存储器中的一个或多个。并行处理单元202外部的任何存储器可被用作全局存储器。在其中处理集群214包括图形多处理器234的多个实例的实施例可共享共同的指令和数据,该共同的指令和数据可被存储在L1缓存248中。
每个处理集群214可包括被配置成用于将虚拟地址映射到物理地址的MMU 245(memory management unit,存储器管理单元)。在其他实施例中,MMU 245的一个或多个实例可驻留在图2A的存储器接口218内。MMU 245包括用于将虚拟地址映射到片的物理地址的页表条目(page table entry,PTE)的集合,并且任选地包括缓存行索引。MMU 245可包括可驻留在处理集群214的图形多处理器234或L1缓存248内的地址转译后备缓冲器(translation lookaside buffer,TLB)或缓存。物理地址被处理,以分发表面数据访问局部性,从而允许分区单元之间的高效的请求交织。缓存行索引可用于确定对缓存行的请求是命中还是未命中。
在图形和计算应用中,处理集群214可被配置成使得每个图形多处理器234耦合至纹理单元236以用于执行纹理映射操作,例如,确定纹理样本位置、读取纹理数据以及过滤纹理数据。纹理数据从内部纹理L1缓存(未示出)中被读取,或者在一些实施例中,从图形多处理器234内的L1缓存中被读取,并且根据需要从L2缓存、本地并行处理器存储器或系统存储器被取得。每个图形多处理器234将经处理的任务输出到数据交叉开关240,以将经处理器的任务提供给另一处理集群214以用于进一步处理,或经由存储器交叉开关216将经处理的任务存储在L2缓存、本地并行处理器存储器或系统存储器中。preROP 242(pre-rasteroperations unit,预先栅格操作单元)被配置成用于从图形多处理器234接收数据,将数据引导至ROP单元,这些ROP单元可与如本文中所描述的分区单元(例如,图2A的分区单元220A-220N)一起被定位。preROP 242单元可针对颜色混合执行优化,组织像素颜色数据,并且执行地址转译。
将领会,本文中所描述的核心体系结构是说明性的,并且变体和修改是可能的。可将任何数量的处理单元(例如,图形多处理器234、纹理单元236、preROP 242等)包括在处理集群214内。进一步地,虽然示出仅一个处理集群214,但是如本文中所描述的并行处理单元可包括处理集群214的任何数量的实例。任选地,每个处理集群214可被配置成用于使用单独且不同的处理单元、L1缓存、L2缓存等来独立于其他处理集群214进行操作。
图2D示出图形多处理器234的示例,其中图形多处理器234与处理集群214的管线管理器232耦合。图形多处理器234具有执行管线,该执行管线包括但不限于指令缓存252、指令单元254、地址映射单元256、寄存器堆258、一个或多个通用图形处理单元(GPGPU)核心262以及一个或多个加载/存储单元266。GPGPU核心262和加载/存储单元266经由存储器和缓存互连268与缓存存储器272和共享存储器270耦合。图形多处理器234可附加地包括张量/或光线追踪核心263,该张量和/或光线追踪核心263包括用于加速矩阵和/或光线追踪操作的硬件逻辑。
指令缓存252可从管线管理器232接收要执行的指令流。指令被缓存在指令缓存252中,并且被调遣以供由指令单元254执行。指令单元254可以将指令作为线程组(例如,单元组(warp))进行调遣,其中,线程组中的每个线程被指派给GPGPU核心262内的不同的执行单元。指令可通过指定统一地址空间内的地址来访问本地地址空间、共享地址空间或全局地址空间中的任一者。可以使用地址映射单元256将统一地址空间中的地址转译为可以由加载/存储单元266访问的不同的存储器地址。
寄存器堆258为图形多处理器234的功能单元提供寄存器的集合。寄存器堆258为连接到图形多处理器234的功能单元(例如,GPGPU核心262、加载/存储单元266)的数据路径的操作对象提供临时存储。寄存器堆258可在功能单元中的每个功能单元之间进行划分,使得每个功能单元被分配寄存器堆258的专用部分。例如,寄存器堆258可在由图形多处理器234执行的不同单元组之间进行划分。
GPGPU核心262可以各自包括用于执行图形多处理器234的指令的浮点单元(floating point unit,FPU)和/或整数算术逻辑单元(arithmetic logic unit,ALU)。在一些实现方式中,GPGPU核心262可包括能以其他方式驻留在张量和/或光线追踪核心263内的硬件逻辑。GPGPU核心262在体系结构上可以是类似的,或者在体系结构上可以是不同的。例如并且在一个实施例中,GPGPU核心262的第一部分包括单精度FPU和整数ALU,而GPGPU核心的第二部分包括双精度FPU。任选地,FPU可实现针对浮点算术的IEEE 754-2008标准,或启用可变精度浮点算术。图形多处理器234可附加地包括用于执行特定功能的一个或多个固定功能或特殊功能单元,该特定功能诸如复制矩形或像素混合操作。GPGPU核心中的一个或多个GPGPU核心还可包括固定功能或特殊功能逻辑。
GPGPU核心262可包括能够对数据的多个集合执行单个指令的SIMD逻辑。任选地,GPGPU核心262可以物理地执行SIMD4、SIMD8和SIMD16指令,并且在逻辑上执行SIMD1、SIMD2和SIMD32指令。针对GPGPU核心的SIMD指令可以由着色器编译器在编译时生成,或在执行针对单程序多数据(single program multipledata,SPMD)或SIMT体系结构而编写并且编译的程序时自动地生成。可以经由单个SIMD指令来执行被配置成用于SIMT执行模型的程序的多个线程。例如并且在一个实施例中,可以经由单个SIMD8逻辑单元来并行地执行八个SIMT线程,这八个SIMT线程执行相同或类似的操作。
存储器和缓存互连268是将图形多处理器234的功能单元中的每个功能单元连接到寄存器堆258并连接至共享存储器270的互连网络。例如,存储器和缓存互连268是允许加载/存储单元266实现共享存储器270与寄存器堆258之间的加载和存储操作的交叉开关互连。寄存器堆258能以与GPGPU核心262相同的频率进行操作,因此GPGPU核心262与寄存器堆258之间的数据传输是非常低等待时间的。共享存储器270可用于启用在图形多处理器234内的功能单元上执行的线程之间的通信。缓存存储器272可被用作数据缓存,例如,以对在功能单元与纹理单元236之间传递的纹理数据进行缓存。共享存储器270还可被用作被管理的经缓存的程序。共享存储器270和缓存存储器272可与数据交叉开关240耦合,以实现与处理集群的其他部件的通信。在GPGPU核心262上执行的线程除了被存储在缓存存储器272内的被自动缓存的数据之外还能以编程方式将数据存储在共享存储器内。
图3A-图3C图示根据实施例的附加的图形多处理器。图3A-图3B图示图形多处理器325、350,图形多处理器325、350与图2C的图形多处理器234相关,并且可替代那些图形多处理器中的一个来被使用。因此,本文中结合图形多处理器234对任何特征的公开也公开了对应的与图形多处理器325、350的结合,但不限于此。图3C图示图形处理单元(GPU)380,该GPU380包括布置为多核心组365A-365N的专用的图形处理资源集合,多核心组365A-365N与图形多处理器325、350对应。所图示的图形多处理器325、350和多核心组365A-365N可以是能够同时执行大量执行线程的流式多处理器(streaming multiprocessors,SM)。
图3A的图形多处理器325包括相对于图2D的图形多处理器234的、执行资源单元的多个附加实例。例如,图形多处理器325可包括指令单元332A-332B、寄存器堆334A-334B和(一个或多个)纹理单元344A-344B的多个实例。图形多处理器325还包括多个图形或计算执行单元集合(例如,GPGPU核心336A-336B、张量核心337A-337B、光线追踪核心338A-338B)以及多个加载/存储单元集合340A-340B。执行资源单元具有共同的指令缓存330、纹理和/或数据缓存存储器342、以及共享存储器346。
各部件可以经由互连结构327进行通信。互连结构327可包括一个或多个交叉开关以实现图形多处理器325的各部件之间的通信。互连结构327是单独的、高速网络结构层,图形多处理器325的每个部件堆叠在该网络结构层上。图形多处理器325的部件经由互连结构327与远程部件通信。例如,核心336A-336B、337A-337B以及338A-338B可以各自经由互连结构327与共享存储器346通信。互连结构327可对图形多处理器325内的通信进行仲裁,以确保部件之间公平的带宽分配。
图3B的图形多处理器350包括多个执行资源集合356A-356D,其中,如图2D和图3A中所图示,每个执行资源集合包括多个指令单元、寄存器堆、GPGPU核心以及加载存储单元。执行资源356A-356D可与用于纹理操作的(一个或多个)纹理单元360A-360D协同地工作,同时共享指令缓存354和共享存储器353。例如,执行资源356A-356D可共享指令缓存354和共享存储器353以及纹理和/或数据缓存存储器358A-358B的多个实例。各部件可经由与图3A的互连结构327类似的互连结构352进行通信。
本领域技术人员将理解,图1、图2A-图2D以及图3A-图3B中所描述的体系结构是描述性的,并且在当前实施例的范围方面不是限制性的。因此,本文中描述的技术可在任何经适当地配置的处理单元上实现而不背离本文中描述的实施例的范围,这些处理单元包括但不限于:一个或多个移动应用处理器;一个或多个桌面型电脑或服务器中央处理单元(central processing unit,CPU),包括多核心CPU;一个或多个并行处理器单元,诸如,图2A的并行处理单元202以及一个或多个图形处理器或专用处理单元。
本文中所描述的并行处理器或GPGPU可通信地耦合至主机/处理器核心以加速图形操作、机器学习操作、模式分析操作以及各种通用GPU(GPGPU)功能。GPU可通过总线或另一互连(例如,高速互连,诸如,PCIe、NVLink或其他已知的协议、标准化协议、或专属协议)通信地耦合至主机处理器/核心。在其他实施例中,GPU可集成在与核心相同的封装或芯片上,并且通过内部处理器总线/互连(即,在封装或芯片内部)通信地耦合至核心。无论GPU被连接所采取的方式如何,处理器核心都可将工作以工作描述符中所包含的命令/指令序列的形式分配给GPU。GPU随后使用专用电路/逻辑来高效地处理这些命令/指令。
图3C图示图形处理单元(GPU)380,该GPU 380包括布置为多核心组365A-365N的专用的图形处理资源集合。虽然提供仅单个多核心组365A的细节,但是将领会,其他多核心组365B-365N可配备有相同或类似的图形处理资源集合。关于多核心组365A-365描述的细节还可应用于本文中描述的任何图形多处理器234、325、350。
如所图示,多核心组365A可包括图形核心的集合370、张量核心的集合371以及光线追踪核心的集合372。调度器/调遣器368调度和调遣图形线程以用于在各个核心370、371、372上执行。寄存器堆的集合369存储在执行图形线程时由核心370、371、372使用的操作对象值。这些寄存器堆可包括例如用于存储整数值的整数寄存器、用于存储浮点值的浮点寄存器、用于存储紧缩(packed)数据元素(整数和/或浮点数据元素)的向量寄存器以及用于存储张量/矩阵值的片寄存器。片寄存器可被实现为向量寄存器的经组合的集合。
一个或多个经组合的第一级(L1)缓存和共享存储器单元373在本地将图形数据存储在每个多核心组365A内,图形数据诸如纹理数据、顶点数据、像素数据、光线数据、包围体数据等。一个或多个纹理单元374也可用于执行纹理操作,诸如,纹理映射和采样。由所有多核心组365A-365N或多核心组365A-365N的子集共享的第二级(L2)缓存375存储用于多个并发的图形线程的图形数据和/或指令。如所图示,可跨多个多核心组365A-365N共享L2缓存375。一个或多个存储器控制器367将GPU 380耦合至存储器366,该存储器366可以是系统存储器(例如,DRAM)和/或专用图形存储器(例如,GDDR6存储器)。
输入/输出(Input/output,I/O)电路363将GPU 380耦合至一个或多个I/O设备362,这一个或多个I/O设备362诸如数字信号处理器(digital signal processor,DSP)、网络控制器或用户输入设备。片上互连可用于将I/O设备362耦合至GPU 380和存储器366。I/O电路363的一个或多个I/O存储器管理单元(I/O memory management unit,IOMMU)364直接将I/O设备362耦合至系统存储器366。任选地,IOMMU 364管理用于将虚拟地址映射到系统存储器366中的物理地址的多个页表集合。I/O设备362、(一个或多个)CPU 361和(一个或多个)GPU 380随后可共享相同的虚拟地址空间。
在IOMMU 364的一个实现方式中,IOMMU 364支持虚拟化。在这种情况下,IOMMU364可以管理用于将宾客/图形虚拟地址映射到宾客/图形物理地址的第一页表集合以及用于将宾客/图形物理地址映射到(例如,系统存储器366内的)系统/主机物理地址的第二页表集合。第一页表集合和第二页表集合中的每一个的基址可被存储在控制寄存器中,并且在上下文切换时被换出(例如,使得新上下文被提供有对相关页表集合的访问权)。虽然未在图3C中图示,但是核心370、371、372和/或多核心组365A-365N中的每一个可包括转译后备缓冲器(TLB),这些TLB用于对宾客虚拟至宾客物理转译、宾客物理至主机物理转译以及宾客虚拟至主机物理转译进行缓存。
(一个或多个)CPU 361、GPU 380和I/O设备362可以被集成在单个半导体芯片和/或芯片封装上。所图示的存储器366可集成在同一芯片上,或者可经由片外接口被耦合至存储器控制器367。在一个实现方式中,存储器366包括共享与其他物理系统级存储器相同的虚拟地址空间的GDDR6存储器,但是本文中描述的基本原理不限于该特定的实现方式。
张量核心371可包括专门被设计成用于执行矩阵操作的多个执行单元,这些矩阵操作是用于执行深度学习操作的基本计算操作。例如,可将同步矩阵乘法操作用于神经网络训练和推断。张量核心371可使用各种操作对象精度来执行矩阵处理,各种操作对象精度包括单精度浮点(例如,32比特)、半精度浮点(例如,16比特)、整数字(16比特)、字节(8比特)和半字节(4比特)。例如,神经网络实现方式提取每个经渲染场景的特征,从而潜在地组合来自多个帧的细节,以构建高质量的最终图像。
在深度学习实现方式中,可调度并行的矩阵乘法工作以用于在张量核心371上执行。神经网络的训练尤其需要大量矩阵点积操作。为了处理N x N x N矩阵乘法的内积公式化,张量核心371可包括至少N个点积处理元件。在矩阵乘法开始之前,一个完整的矩阵被加载到片寄存器中,并且对于N个循环中的每个循环,第二矩阵的至少一列被加载。对于每个循环,存在被处理的N个点积。
取决于特定的实现方式,能以不同精度来存储矩阵元素,包括16比特的字、8比特的字节(例如,INT8)以及4比特的半字节(例如,INT4)。可为张量核心371指定不同的精度模式以确保将最高效的精度用于不同的工作负载(例如,诸如推断工作负载,其可容忍至字节和半字节的量化(quantization))。所支持的格式附加地包括64比特浮点(64-bitfloating point,FP64)和非IEEE浮点格式,诸如,bfloat16格式(例如,Brain浮点)、具有一个符号比特、八个指数比特和八个有效数字比特(其中的七个被显式地存储)的16比特浮点格式。一个实施例包括对降低精度的张量浮点(TF32)模式的支持,该TF32模式使用FP32(8比特)的范围和FP16(10比特)的精度执行计算。能以相对于FP32更高的性能以及相对于FP16增加的精度对FP32输入执行降低精度的TF32操作并产生FP32输出。在一个实施例中,支持一个或多个8比特浮点格式(FP32)。
在一个实施例中,张量核心371支持用于在其中绝大多数值为零的矩阵的稀疏操作模式。张量核心371包括对以稀疏矩阵表示(例如,坐标列表编码(COO)、压缩稀疏行(CSR)、压缩稀疏列(CSC)等)来编码的稀疏输入矩阵的支持。张量核心371还包括对在稀疏矩阵表示可被进一步压缩的情况下的经压缩的稀疏矩阵表示的支持。经压缩的矩阵数据、经编码的矩阵数据和/或经压缩且经编码的矩阵数据以及相关联的压缩和/或编码元数据可由张量核心371读取,并且非零值可被提取。例如,对于给定的输入矩阵A,非零值可从矩阵A的至少部分的经压缩的和/或经编码的表示来加载。基于矩阵A中非零值的位置(其可从与非零值相关联的索引或坐标元数据确定),输入矩阵B中的对应值可被加载。取决于要执行的操作(例如,乘法),如果对应的值是零值,则从输入矩阵B加载值可被绕过。在一个实施例中,对于某些操作(诸如,乘法操作)的值的配对可由调度器逻辑预扫描,并且仅非零输入之间的操作被调度。取决于矩阵A和矩阵B的维度以及要执行的操作,输出矩阵C可以是密集或稀疏的。在输出矩阵C是稀疏的情况下且取决于张量核心371的配置,输出矩阵C可以按压缩格式、稀疏编码或压缩稀疏编码被输出。
光线追踪核心372可加速用于实时光线追踪实现方式和非实时光线追踪实现方式两者的光线追踪操作。具体而言,光线追踪核心372可包括光线遍历/相交电路,该光线遍历/相交电路用于使用包围体层次体系(bounding volume hierarchy,BVH)来执行光线遍历并标识封围在BVH体积内的光线与基元之间的相交。光线追踪核心372还可包括用于执行深度测试和剔除(例如,使用Z缓冲器或类似布置)的电路。在一个实现方式中,光线追踪核心372与本文中描述的图像降噪技术协同地执行遍历和相交操作,该图像降噪技术的至少部分可在张量核心371上执行。例如,张量核心371可实现深度学习神经网络以执行对由光线追踪核心372生成的帧的降噪。然而,(一个或多个)CPU 361、图形核心370和/或光线追踪核心372还可实现全部的降噪和/或深度学习算法或降噪和/或深度学习算法中的部分。
此外,如上文所描述,可采用对于降噪的分布式方法,其中,GPU 380在通过网络或高速互连而耦合至其他计算设备的计算设备中。按照该分布式方法,经互连的计算设备可共享神经网络学习/训练数据,以改善整个系统学习执行用于不同类型的图像帧和/或不同的图形应用的降噪的速度。
光线追踪核心372可处理所有的BVH遍历和/或光线-基元相交,从而使图形核心370免于被针对每条光线的数千个指令过载。例如,每个光线追踪核心372包括用于执行包围盒测试(例如,用于遍历操作)的第一专业电路集合和/或用于执行光线-三角形相交测试(例如,使已被遍历的光线相交)的第二专业电路集合。因此,例如,多核心组365A可简单地启动光线探测,并且光线追踪核心372独立地执行光线遍历和相交,并将命中数据(例如,命中、无命中、多个命中等)返回到线程上下文。当光线追踪核心370执行遍历和相交操作时,其他核心371、372被释放以执行其他图形或计算工作。
任选地,每个光线追踪核心372可包括用于执行BVH测试操作的遍历单元和/或执行光线-基元相交测试的相交单元。相交单元生成“命中”、“无命中”或“多个命中”响应,该相交单元将这些响应提供给适当的线程。在遍历和相交操作期间,其他核心(例如,图形核心370和张量核心371)的执行资源被释放以执行其他形式的图形工作。
在下文描述的一个任选实施例中,使用在其中工作被分布在图形核心370与光线追踪核心372之间的混合式栅格化/光线追踪方法。
光线追踪核心372(和/或其他核心370、371)可包括对光线追踪指令集的硬件支持,光线追踪指令集诸如:微软的DirectX光线追踪(DirectX Ray Tracing,DXR),其包括DispatchRays命令;以及光线生成着色器、最近命中着色器、任何命中着色器和未命中着色器,它们使得能够为每个对象指派唯一的着色器和纹理集合。可由光线追踪核心372、图形核心370和张量核心371支持的另一光线追踪平台是Vulkan API(例如,Vulkan版本1.1.85,或者更晚的版本)。然而,要注意,本文中描述的基本原理不限于任何特定的光线追踪ISA。
一般而言,各个核心372、371、370可支持包括用于以下各项中的一项或多项的指令/函数的光线追踪指令集:光线生成、最近命中、任何命中、光线-基元相交、逐基元和层次体系包围盒构建、未命中、拜访和异常。更具体地,优选的实施例包括用于执行以下功能中的一项或多项的光线追踪指令:
光线生成——可为每个像素、样本或其他用户定义的工作指派执行光线生成指令。
最近命中——可执行最近命中指令以对场景内光线与基元的最近交点定位。
任何命中——任何命中指令标识场景内光线与基元之间的多个相交,从而潜在地标识新的最近交点。
相交——相交指令执行光线-基元相交测试并输出结果。
逐基元包围盒构建——该指令围绕给定的基元或基元组建立包围盒(例如,当建立新BVH或其他加速数据结构时)。
未命中——指示光线未命中场景或场景的指定区域内的所有几何体。
拜访——指示光线将遍历的子容体。
异常——包括各种类型的异常处置器(例如,针对各种错误条件被调用)。
在一个实施例中,光线追踪核心372可适于加速通用计算操作,这些通用计算操作可使用与光线相交测试类似的计算技术来加速。可提供计算框架,该计算框架使着色器程序能够被编译为经由光线追踪核心执行通用计算操作的低级别指令和/或基元。可受益于在光线追踪核心372上执行的计算操作的示例性计算问题包括涉及坐标空间内光束、波、光线或粒子传播的计算。可相对于坐标空间内的几何体或网格计算与那个传播相关联的交互。例如,与通过环境的电磁信号传播相关联的计算可经由使用经由光线追踪核心被执行的指令或基元来加速。信号通过环境中的对象发生的折射和反射可被计算为直接的光线追踪模拟。
光线追踪核心372还可用于执行不直接与光线追踪类似的计算。例如,可使用光线追踪核心372来加速网格投影、网格细化和体积采样计算。还可执行通用坐标空间计算,诸如,最近邻计算。例如,可通过定义坐标空间中围绕给定点的包围盒来发现该点附近的点的集合。随后可使用光线追踪核心372内的BVH和光线探测逻辑来确定包围盒内点相交的集合。相交构成原点以及那个原点的最近邻。可并行于在图形核心372和张量核心371上执行的计算来执行使用光线追踪核心372执行的计算。着色器编译器可被配置成用于将计算着色器或其他通用图形处理程序编译为能够跨图形核心370、张量核心371和光线追踪核心372被并行化的低级别基元。
用于GPU至主机处理器互连的技术
图4A图示在其中多个GPU 410-413(例如,诸如图2A中示出的并行处理器200)通过高速链路440A-440D(例如,总线、点到点互连等)通信地耦合至多个多核心处理器405-406的示例性体系结构。取决于实现方式,高速链路440A-440D可支持4GB/s、30GB/s、80GB/s或更高的通信吞吐量。可使用各种互连协议,包括但不限于PCIe 4.0或5.0和NVLink 2.0。然而,本文中描述的基本原理不限于任何特定的通信协议或吞吐量。
GPU 410-413中的两个或更多个可通过高速链路442A-442B被互连,高速链路442A-442B可使用与针对高速链路440A-440D使用的那些协议/链路相同或不同的协议/链路来实现。类似地,多核心处理器405-406中的两个或更多个可通过高速链路443进行连接,该高速链路443可以是在20GB/s、30GB/s、120GB/s或者更低或更高速度下进行操作的对称多处理器(symmetric multi-processor,SMP)总线。替代地,图4A中示出的各种系统部件之间的所有通信可使用相同的协议/链路(例如,通过共同的互连结构)来实现。然而,如所提及,本文中描述的基本原理不限于任何特定类型的互连技术。
多核心处理器405和多核心处理器406中的每一个可分别经由存储器互连430A-430B通信地耦合至处理器存储器401-402,并且每个GPU 410-413分别通过GPU存储器互连450A-450D通信地耦合至GPU存储器420-423。存储器互连430A-430B和450A-450D可利用相同或不同的存储器访问技术。作为示例并且不作为限制,处理器存储器401-402和GPU存储器420-423可以是诸如动态随机存取存储器(DRAM)(包括堆叠式DRAM)、图形DDR SDRAM(GDDR)(例如,GDDR5、GDDR6)、或高带宽存储器(HBM)之类的易失性存储器,并且/或者可以是诸如3DXpoint/Optane或Nano-Ram之类的非易失性存储器。例如,存储器的某个部分可以是易失性存储器,并且另一部分可以是非易失性存储器(例如,使用两级存储器(two-levelmemory,2LM)层次体系)。本文中描述的存储器子系统可与数种存储器技术兼容,这些存储器技术诸如由JEDEC(Joint Electronic Device Engineering Council,联合电子设备工程委员会)发布的双倍数据速率版本。
如下文所描述,尽管各处理器405-406和GPU 410-413可分别物理地耦合至特定的存储器401-402、420-423,但是可实现在其中同一虚拟系统地址空间(也被称为“有效地址”空间)在所有的各种物理存储器之间分布的统一存储器体系结构。例如,处理器存储器401-402各自可包括64GB的系统存储器地址空间,并且GPU 420-423各自可包括32GB的系统存储器地址空间(从而在该示例中产生总共256GB的可寻址存储器)。
图4B图示多核心处理器407与图形加速模块446之间的互连的附加的任选细节。图形加速模块446可包括集成在线卡上的一个或多个GPU芯片,该线卡经由高速链路440耦合至处理器407。替代地,图形加速模块446可集成在与处理器407相同的封装或芯片上。
所图示的处理器407包括多个核心460A-460D,这些核心各自具有转译后备缓冲器461A-461D以及一个或多个缓存462A-462D。核心可包括用于执行指令并处理数据的各种其他部件,未图示出这些部件以避免使本文中描述的部件(例如,指令取得单元、分支预测单元、解码器、执行单元、重排序缓冲器等)的基本原理模糊。缓存462A-462D可包括第一级(L1)缓存和第二级(L2)缓存。此外,一个或多个共享缓存456可被包括在缓存层次体系中,并由核心的集合460A-460D共享。例如,处理器407的一个实施例包括24个核心,这些核心各自具有其自身的L1缓存、十二个共享L2缓存以及十二个共享L3缓存。在该实施例中,L2缓存和L3缓存中的一者由两个相邻的核心共享。处理器407和图形加速器集成模块446与系统存储器441连接,该系统存储器441可包括处理器存储器401-402。
经由通过一致性总线464的核心间通信为存储在各缓存462A-462D、456和系统存储器441中的数据和指令维持一致性。例如,每个缓存可具有与其相关联的缓存一致性逻辑/电路,以响应于检测到的对特定缓存行的读取或写入而通过一致性总线464进行通信。在一个实现方式中,通过一致性总线464实现缓存监听协议,以监听缓存访问。缓存监听/一致性技术为本领域技术人员很好地理解,并且将不在此详细描述,以避免使本文中描述的基本原理模糊。
可以提供将图形加速模块446通信地耦合至一致性总线464的代理电路425,从而允许图形加速模块446作为核心的对等方参与缓存一致性协议。具体而言,接口435提供通过高速链路440(例如,PCIe总线、NVLink等)而至代理电路425的连接性,并且接口437将图形加速模块446连接到高速链路440。
在一个实现方式中,加速器集成电路436代表图形加速模块446的多个图形处理引擎431、432…N提供缓存管理、存储器访问、上下文管理以及中断管理服务。图形处理引擎431、432…N各自可包括分开的图形处理单元(GPU)。替代地,图形处理引擎431、432…N可包括GPU内的不同类型的图形处理引擎,诸如,图形执行单元、媒体处理引擎(例如,视频编码器/解码器)、采样器以及块图像传输(block image transfer,BLIT)引擎。换言之,图形加速模块可以是具有多个图形处理引擎431-432…N的GPU,或者图形处理引擎431-432…N可以是集成在共同的封装、线卡或芯片上的单独的GPU。图形处理引擎431-432…N可以利用本文描述的任何图形处理器或计算加速器体系结构来配置。
加速器集成电路436可包括存储器管理单元(MMU)439,该MMU 439用于执行诸如虚拟到物理存储器转译(也称为有效到实际存储器转译)之类的各种存储器管理功能以及用于访问系统存储器441的存储器访问协议。MMU 439还可包括用于对虚拟/有效到物理/实际地址转译进行缓存的转译后备缓冲器(TLB)(未示出)。在一个实现方式中,缓存438存储用于由图形处理引擎431、432…N高效访问的命令和数据。存储在缓存438和图形存储器433-434…M中的数据可被保持与核心缓存462A-462D、456和系统存储器441一致。如所提及,这可经由代理电路425来完成,该代理电路425代表缓存438和存储器433-434…M参与缓存一致性机制(例如,向缓存438发送与处理器缓存462A-462D、456上的缓存行的修改/访问相关的更新,并从缓存438接收更新)。
寄存器的集合445存储用于由图形处理引擎431-432…N执行的线程的上下文数据,并且上下文管理电路448管理这些线程上下文。例如,上下文管理电路448可执行保存和恢复操作,以在上下文切换期间保存和恢复各线程的上下文(例如,其中第一线程被保存并且第二线程被恢复,使得第二线程可由图形处理引擎执行)。例如,在上下文切换时,上下文管理电路448可将当前寄存器值存储到存储器中的指定区域(例如,由上下文指针标识)。当返回到该上下文时,其随后可恢复寄存器值。中断管理电路447例如可从系统设备接收中断,并处理从系统设备接收的中断。
在一个实现方式中,由MMU 439将来自图形处理器引擎431的虚拟/有效地址转译为系统存储器441中的实际/物理地址。任选地,加速器集成电路436支持多个(例如,4个、8个、16个)图形加速器模块446和/或其他加速器设备。图形加速器模块446可专用于在处理器407上执行的单个应用,或者可在多个应用之间被共享。任选地,提供虚拟化图形执行环境,在该虚拟化图形执行环境中,图形处理引擎431-432…N的资源与多个应用、虚拟机(virtual machine,VM)或容器共享。资源可被细分为“切片”,这些切片基于与VM和/或应用相关联的处理要求和优先级或基于用于图形加速器模块446的预定分区简档而被分配给不同的VM和/或应用。VM和容器在本文中可以可互换地使用。
虚拟机(VM)可以是运行操作系统以及一个或多个应用的软件。VM可由规范、配置文件、虚拟盘文件、非易失性随机存取存储器(non-volatile random access memory,NVRAM)设置文件和日志文件来定义,并且由主机计算平台的物理资源来备份。VM可包括被安装在模仿专用硬件的软件上的操作系统(operating system,OS)或应用环境。终端用户在虚拟机上具有如同他们将在专用硬件上所具有的相同体验。被称为管理程序的专业化软件完全对PC客户端或服务器的CPU、存储器、硬盘、网络和其他硬件资源进行仿真,从而使虚拟机能够共享资源。管理程序可对彼此隔离的多个虚拟硬件平台进行仿真,从而允许虚拟机在相同的底层物理主机上运行Server、VMware ESXi和其他操作系统。
容器可以是应用、配置和依赖关系的软件包,因此,应用在一个计算环境到另一计算环境可靠地运行。容器可共享安装在服务器平台上的操作系统,并且作为隔离的进程运行。容器可以是包含软件运行所需的任何内容(诸如,系统工具、库和设置)的软件包。容器并不像传统的软件程序那样被安装,这允许容器与其他软件隔离且与操作系统自身隔离。容器的隔离性质提供若干益处。首先,容器中的软件将在不同环境中以相同方式运行。例如,包括PHP和MySQL的容器可在计算机和机器两者上以完全相同的方式运行。其次,容器提供增加的安全性,因为软件将不会影响主机操作系统。虽然安装的应用会更改系统设置并修改资源(诸如,Windows注册表),但是容器能够仅修改该容器内的设置。
因此,加速器集成电路436充当用于图形加速模块446的、到系统的桥接器,并且提供地址转译和系统存储器缓存服务。在一个实施例中,为了促进桥接功能,加速器集成电路436还可包括共享I/O 497(例如,PCIe、USB或其他元件)和硬件,以实现对电压、时钟控制、性能、热和安全性的系统控制。共享I/O 497可利用分开的物理连接,或者可跨越高速链路440。此外,加速器集成电路436可提供虚拟化设施,以供主机处理器管理对图形处理引擎、中断和存储器管理的虚拟化。
由于图形处理引擎431-432…N的硬件资源被显式地映射到由主机处理器407看到的实际地址空间,因此任何主机处理器可以使用有效地址值来直接对这些资源进行寻址。加速器集成电路436的一个任选功能是对图形处理引擎431-432…N进行物理分离,使得它们对于系统显得像独立的单元。
一个或多个图形存储器433-434…M可分别耦合至图形处理引擎431-432…N中的每一个。图形存储器433-434…M存储由图形处理引擎431-432…N中的每一个处理的指令和数据。图形存储器433-434…M可以是诸如DRAM(包括堆叠式DRAM)、GDDR存储器(例如,GDDR5、GDDR6)或HBM之类的易失性存储器,并且/或者可以是诸如3DXpoint/Optane、三星Z-NAND、或Nano-Ram之类的非易失性存储器。
为了减少高速链路440上的数据通信量,可使用偏置技术来确保存储在图形存储器433-434…M中的数据是将由图形处理引擎431-432…N最频繁地使用并且优选地不由核心460A-460D使用(至少不频繁地使用)的数据。类似地,偏置机制尝试将核心(并且优选地不是图形处理引擎431-432…N)所需要的数据保持在系统存储器441和核心的缓存462A-462D、456内。
根据图4C中示出的变体,加速器集成电路436被集成在处理器407内。图形处理引擎431-432…N通过高速链路440、经由接口437和接口435(其再次可利用任何形式的总线或接口协议)直接向加速器集成电路436通信。加速器集成电路436可执行与关于图4B所描述的那些操作相同的操作,但是考虑到该加速器集成电路436与一致性总线464和缓存462A-462D、456紧密邻近,其潜在地能以较高的吞吐量执行操作。
所描述的实施例可支持不同的编程模型,这些编程模型包括专用进程编程模型(无图形加速模块虚拟化)和共享编程模型(具有虚拟化)。后者可包括受加速器集成电路436控制的编程模型以及受图形加速模块446控制的编程模型。
在专用进程模型的实施例中,图形处理引擎431、432、…、N在单个操作系统下可专用于单个应用或进程。单个应用可使其他应用请求漏到图形引擎431、432、…、N,从而在VM/分区内提供虚拟化。
在专用进程编程模型中,图形处理引擎431、432…N可由多个VM/应用分区共享。共享模型要求系统管理程序使图形处理引擎431-432…N虚拟化,以允许由每个操作系统进行的访问。对于不具有管理程序的单分区系统,图形处理引擎431-432…N由操作系统拥有。在这两种情况下,操作系统可使图形处理引擎431-432…N虚拟化,以提供对每个进程或应用的访问权。
对于共享编程模型,图形加速模块446或各个图形处理引擎431-432…N使用进程句柄来选择进程要素。进程要素可被存储在系统存储器441中,并且可以是使用本文中所描述的有效地址到实际地址转译技术而可寻址的。进程句柄可以是在向图形处理引擎431-432…N注册其上下文(即,调用系统软件以将进程要素添加到进程要素链表)时提供给主机进程的实现方式特定的值。进程句柄的较低的16比特可以是进程要素在进程要素链表内的偏移。
图4D图示示例性加速器集成切片490。如本文中所使用,“切片”包括加速器集成电路436的处理资源的指定部分。系统存储器441内的应用有效地址空间482存储进程要素483。进程要素483可响应于来自在处理器407上执行的应用480的GPU调用481而被存储。进程要素483包含对应应用480的进程状态。包含在进程要素483中的工作描述符(workdescriptor,WD)484可以是由应用请求的单个作业,或者可包含指向作业队列的指针。在后一种情况下,WD 484是指向应用的地址空间482中的作业请求队列的指针。
图形加速模块446和/或各个图形处理引擎431-432…N可以由系统中的全部进程或系统中的进程的子集共享。例如,本文中描述的技术可包括用于建立进程状态并将WD484发送至图形加速模块446以在虚拟化环境中开始作业的基础设施。
在一个实现方式中,专用进程编程模型是实现方式特定的。在该模型中,单个进程拥有图形加速模块446或单独的图形处理引擎431。由于图形加速模块446由单个进程拥有,因此图形加速模块446被指派时,管理程序针对拥有的分区对加速器集成电路436进行初始化,并且操作系统针对拥有的进程对加速器集成电路436进行初始化。
在操作中,加速器集成切片490中的WD取得单元491取得下一WD 484,该下一WD484包括要由图形加速模块446的图形处理引擎中的一个图形处理引擎完成的工作的指示。如所图示,来自WD 484的数据可被存储在寄存器445中,并且由MMU 439、中断管理电路447和/或上下文管理电路448使用。例如,MMU 439可包括用于访问OS虚拟地址空间485内的段表/页表486的段/页走查电路。中断管理电路447可处理从图形加速模块446接收的中断事件492。当执行图形操作时,由图形处理引擎431-432…N生成的有效地址493由MMU 439转译成实际地址。
相同的寄存器集合445针对每个图形处理引擎431-432…N和/或图形加速模块446可被复制,并且可由管理程序或操作系统初始化。这些被复制的寄存器中的每个寄存器可被包括在加速器集成切片490中。在一个实施例中,每个图形处理引擎431-432…N可作为不同的图形处理器设备向管理程序496呈现。可针对特定图形处理引擎431-432…N的客户端配置QoS设置,并且可启用每个引擎的客户端之间的数据隔离。在表1中示出可由管理程序初始化的示例性寄存器。
表1-管理程序初始化的寄存器
1 切片控制寄存器
2 实际地址(Real Address,RA)调度的进程区域指针
3 权限掩码覆盖寄存器
4 中断向量表条目偏移
5 中断向量表条目限制
6 状态寄存器
7 逻辑分区ID
8 实际地址(RA)管理程序加速器利用记录指针
9 存储描述寄存器
在表2中示出可由操作系统初始化的示例性寄存器。
表2-操作系统初始化的寄存器
1 进程和线程标识
2 有效地址(Effective Address,EA)上下文保存/恢复指针
3 虚拟地址(Virtual Address,VA)加速器利用记录指针
4 虚拟地址(VA)存储段表指针
5 权限掩码
6 工作描述符
每个WD 484对于特定的图形加速模块446和/或图形处理引擎431-432...N可以是特定的。它包含图形处理引擎431-432...N完成其工作所需要的所有信息,或者它可以是指向应用已经建立要完成的工作的命令队列所在的存储器位置的指针。
图4E图示共享模型的附加的任选细节。它包括进程要素列表499被存储在其中的管理程序实际地址空间498。管理程序实际地址空间498是经由管理程序496可访问的,管理程序496针对操作系统495使图形加速模块引擎虚拟化。
共享编程模型允许来自系统中的全部分区或系统中的分区的子集的全部进程或进程的子集使用图形加速模块446。存在在其中图形加速模块446由多个进程和分区共享的两种编程模型:时分共享和图形定向共享。
在该模型中,系统管理程序496拥有图形加速模块446,并使其功能对所有的操作系统495可用。为使图形加速模块446支持由系统管理程序496进行的虚拟化,图形加速模块446可遵守以下要求:1)应用的作业请求必须是自主的(即,状态不需要在作业之间被维持),或者图形加速模块446必须提供上下文保存和恢复机制。2)由图形加速模块446保证应用的作业请求在所指定的时间量内完成,包括任何转换错误,或者图形加速模块446提供抢占对作业的处理的能力。3)图形加速模块446当在定向共享编程模型中操作时必须被保证进程之间的公平性。
对于定向共享模型,可需要应用480来用图形加速模块446类型、工作描述符(WD)、权限掩码寄存器(authority mask register,AMR)值和上下文保存/恢复区域指针(context save/restore area pointer,CSRP)作出操作系统495系统调用。图形加速模块446类型描述针对系统调用的目标加速功能。图形加速模块446类型可以是系统特定值。特别针对图形加速模块446对WD进行格式化,并且WD可以采用以下形式:图形加速模块446命令、指向用户定义的结构的有效地址指针、指向命令队列的有效地址指针、或用于描述要由图形加速模块446完成的工作的任何其他数据结构。在一个实施例中,AMR值是要用于当前进程的AMR状态。值被传递到操作系统与应用设置AMR类似。如果加速器集成电路436和图形加速模块446实现方式不支持用户权限掩码覆盖寄存器(User Authority Mask OverrideRegister,UAMOR),则操作系统可在管理程序调用中传递AMR之前将当前UAMOR值应用到AMR值。管理程序496可在将AMR放置到进程要素483中之前任选地应用当前权限掩码覆盖寄存器(Authority Mask Override Register,AMOR)值。CSRP可以是包含应用的地址空间482中的区域的有效地址以供图形加速模块446用来保存和恢复上下文状态的寄存器445中的一个寄存器。如果没有状态被要求以被保存在作业之间或者当作业被抢占时,则该指针是任选的。上下文保存/恢复区域可以是钉置(pinned)的系统存储器。
在接收到系统调用时,操作系统495可验证应用480已注册并且已被给予使用图形加速模块446的权限。操作系统492随后利用表3中示出的信息来调用管理程序496。
表3-OS对管理程序的调用参数
在接收管理程序调用时,管理程序496验证操作系统495已注册并且已被给予使用图形加速模块446的权限。管理程序496随后将进程要素483置于针对对应的图形加速模块446类型的进程要素链表中。进程要素可包括表4中示出的信息。
表4-进程要素信息
1 工作描述符(WD)
2 权限掩码寄存器(AMR)值(潜在地被掩蔽)。
3 有效地址(EA)上下文保存/恢复区域指针(CSRP)
4 进程ID(PID)和任选的线程ID(TID)
5 虚拟地址(VA)加速器利用记录指针(AURP)
6 存储段表指针(SSTP)的虚拟地址
7 逻辑中断服务号(LISN)
8 中断向量表,从管理程序调用参数推导出。
9 状态寄存器(state register,SR)值
10 逻辑分区ID(logical partition ID,LPID)
11 实阡地址(RA)管理程序加速器利用记录指针
12 存储描述符寄存器(Storage Descriptor Register,SDR)
管理程序可初始化多个加速器集成切片490的寄存器445。
如图4F中所图示,在一个任选实现方式中,采用经由共同的虚拟存储器地址空间可寻址的统一存储器,该共同的虚拟存储器地址空间被用于访问物理处理器存储器401-402和GPU存储器420-423。在该实现方式中,在GPU 410-413上执行的操作利用同一虚拟/有效存储器地址空间来访问处理器存储器401-402并且反之亦然,由此简化可编程性。虚拟/有效地址空间的第一部分可被分配给处理器存储器401,第二部分可被分配给第二处理器存储器402,第三部分可被分配给GPU存储器420,以此类推。整个虚拟/有效存储器空间(有时被称为有效地址空间)由此可跨处理器存储器401-402和GPU存储器420-423中的每一个分布,从而允许任何处理器或GPU利用映射到任一物理存储器的虚拟地址来访问该物理存储器。
可提供MMU 439A-439E中的一个或多个MMU内的偏置/一致性管理电路494A-494E,这些偏置/一致性管理电路494A-494E确保主机处理器(例如,405)的缓存与GPU 410-413的缓存之间的缓存一致性,并且实现指示某些类型的数据应当被存储在其中的物理存储器的偏置技术。虽然在图4F中图示偏置/一致性管理电路494A-494E的多个实例,但是可在一个或多个主机处理器405的MMU内和/或在加速器集成电路436内实现偏置/一致性电路。
GPU附连的存储器420-423可被映射为系统存储器的部分并使用共享虚拟存储器(shared virtual memory,SVM)技术来访问,但是不遭受与完全系统缓存一致性相关联的典型性能缺陷。GPU附连的存储器420-423在没有繁重的缓存一致性开销的情况下作为系统存储器被访问的能力为GPU迁移提供了有益的操作环境。该布置允许主机处理器405在没有传统的I/O DMA数据复制的开销的情况下设置操作对象并访问计算结果。此类传统复制涉及驱动器调用、中断、以及存储器映射的I/O(memory mapped I/O,MMIO)访问,它们相对于简单的存储器访问全都是低效的。同时,在没有缓存一致性开销的情况下访问GPU附连的存储器420-423的能力对于被迁移的计算的执行时间可以是关键的。例如,在具有大量流式写入存储器通信量的情况下,缓存一致性开销可能显著地降低由GPU 410-413看到的有效写入带宽。操作对象设置的效率、结果访问的效率以及GPU计算的效率在确定GPU迁移的有效性时全都发挥作用。
GPU偏置与主机处理器偏置之间的选择可由偏置跟踪器数据结构驱动。例如,可使用偏置表,该偏置表可以是页粒度的结构(即,以存储器页的粒度受控制),该页粒度的结构包括每GPU附连的存储器页的1个或2个比特。偏置表可在一个或多个GPU附连的存储器420-423的偷取的存储器范围中实现,在GPU 410-413中具有或不具有偏置缓存(例如,用于对偏置表的频繁/最近使用的条目进行缓存)。替代地,整个偏置表可被维持在GPU内。
在一个实现方式中,在对GPU存储器的实际访问之前,访问与对GPU附连的存储器420-423的每次访问相关联的偏置表条目,从而导致下列操作。首先,来自GPU 410-413的、在GPU偏置中发现它们的页的本地请求直接被转发至对应的GPU存储器420-423。来自GPU的、在主机偏置中发现它们的页的本地请求被转发至处理器405(例如,如上文所讨论,通过高速链路)。任选地,来自处理器405的、在主机处理器偏置中发现所请求的页的请求像正常存储器读取那样完成该请求。替代地,涉及GPU偏置的页的请求可被转发至GPU 410-413。如果GPU当前不是正在使用该页,则GPU随后可将该页转变为主机处理器偏置。
可通过基于软件的机制、硬件辅助的基于软件的机制或对于有限的情况集合而言通过基于纯硬件的机制来改变页的偏置状态。
一种用于改变偏置状态的机制采用API调用(例如,OpenCL),其进而调用GPU的设备驱动器,该设备驱动器进而向GPU发送引导该GPU改变偏置状态并针对一些转变在主机中执行缓存转储清除操作的消息(或使命令描述符入列)。针对从主机处理器405偏置到GPU偏置的转变需要缓存转储清除操作,但针对相反的转变不需要缓存转储清除操作。
可通过临时渲染不可由主机处理器405缓存的GPU偏置的页来维持缓存一致性。为了访问这些页,处理器405可请求从GPU 410访问,取决于实现方式,GPU 410可以或可以不准予立即访问。因此,为了减少主机处理器405与GPU 410之间的通信,确保GPU偏置的页是GPU所需但不是主机处理器405所需的那些页是有益的,并且反之亦然。
图形处理管线
图5图示图形处理管线500。图形多处理器(诸如,图2D中的图形多处理器234、图3A的图形多处理器325、图3B的图形多处理器350)可实现所图示的图形处理管线500。图形多处理器可被包括在如本文中所描述的并行处理子系统内,并行处理子系统诸如图2A的并行处理器200,其可与图1的(一个或多个)并行处理器112相关并且可替代那些并行处理器中的一个被使用。各种并行处理器系统可经由如本文中所描述的并行处理单元(例如,图2A的并行处理单元202)的一个或多个实例来实现图形处理管线500。例如,着色器单元(例如,图2C的图形多处理器234)可被配置成用于执行以下一者或多者的功能:顶点处理单元504、曲面细分控制处理单元508、曲面细分评估处理单元512、几何处理单元516、以及片段/像素处理单元524。数据组装器502、基元组装器506、514、518、曲面细分单元510、栅格化器522以及栅格操作单元526的功能也可由处理集群(例如,图2A的处理集群214)内的其他处理引擎和对应的分区单元(例如,图2A的分区单元220A-220N)来执行。图形处理管线500还可使用用于一个或多个功能的专用处理单元来实现。图形处理管线500的一个或多个部分由通用处理器(例如,CPU)内的并行处理逻辑执行也是可能的。任选地,图形处理管线500的一个或多个部分可以经由存储器接口528来访问片上存储器(例如,如图2A中的并行处理器存储器222),该存储器接口528可以是图2A的存储器接口218的实例。图形处理器管线500还可经由如图3C中的多核心组365A来实现。
数据组装器502是可收集针对表面和基元的顶点数据的处理单元。数据组装器502随后将顶点数据输出到顶点处理单元504,该顶点数据包括顶点属性。顶点处理单元504是可编程执行单元,该可编程执行单元执行顶点着色器程序,从而按照顶点着色器程序所指定地来照明以及变换顶点数据。顶点处理单元504读取存储在缓存、本地或系统存储器中的数据以供在处理顶点数据时使用,并且可被编程为用于将顶点数据从基于对象的坐标表示变换为世界空间坐标空间或归一化设备坐标空间。
基元组装器506的第一实例从顶点处理单元504接收顶点属性。基元组装器506根据需要读取所存储的顶点属性,并且构建图形基元以供由曲面细分控制处理单元508进行处理。图形基元包括如由各种图形处理应用编程接口(application programminginterface,API)支持的三角形、线段、点、补片等。
曲面细分控制处理单元508将输入顶点视为几何补片的控制点。将控制点从来自补片的输入表示(例如,补片的基础)变换为适合于在由曲面细分评估处理单元512进行的表面评估中使用的表示。曲面细分控制处理单元508还可计算几何补片的边(edge)的曲面细分因子。曲面细分因子应用于单个边,并对与该边相关联的依赖于视图的细节级别进行量化。曲面细分单元510被配置成用于接收补片的边的曲面细分因子,并且用于将补片曲面细分成多个几何基元(诸如,线、三角形或四边形基元),这些几何基元被传送到曲面细分评估处理单元512。曲面细分评估处理单元512对经细分的补片的参数化坐标进行操作,以生成与几何基元相关联的每个顶点的表面表示和顶点属性。
基元组装器514的第二实例根据需要从曲面细分评估处理单元512接收顶点属性,读取所存储的顶点属性,并且构建图形基元以供几何处理单元516进行处理。几何处理单元516是可编程执行单元,该可编程执行单元执行几何着色器程序以按照几何着色器程序所指定地变换从基元组装器514接收到的图形基元。几何处理单元516可被编程为用于将图形基元细分为一个或多个新的图形基元并计算被用于对这些新的图形基元进行栅格化的参数。
几何处理单元516可以能够在几何流中添加或删除元素。几何处理单元516向基元组装器518输出指定新的图形基元的参数和顶点。基元组装器518从几何处理单元516接收参数和顶点,并构建图形基元以供视口缩放、剔除和裁剪单元520进行处理。几何处理单元516读取被存储在并行处理器存储器或系统存储器中的数据,以供在处理几何数据时使用。视口缩放、剔除和裁剪单元520执行裁剪、剔除和视口缩放,并且将经处理的图形基元输出到栅格化器522。
栅格化器522可执行深度剔除和其他基于深度的优化。栅格化器522还对新的图形基元执行扫描转换以生成片段并将那些片段和相关联的覆盖数据输出到片段/像素处理单元524。片段/像素处理单元524是被配置成用于执行片段着色器程序或像素着色器程序的可编程执行单元。片段/像素处理单元524按照片段或像素着色器程序所指定地变换从栅格化器522接收的片段或像素。例如,片段/像素处理单元524可被编程为用于执行包括但不限于纹理映射、着色、混合、纹理校正和透视校正的操作,以产生被输出到栅格操作单元526的经着色的片段或像素。片段/像素处理单元524可读取被存储在并行处理器存储器或系统存储器中的数据,以供在处理片段数据时使用。片段或像素着色器程序可被配置成用于取决于为处理单元配置的采样率而以样本、像素、片或其他粒度进行着色。
栅格操作单元526是处理单元,该处理单元执行栅格操作并将像素数据输出为要被存储在图形存储器(例如,如图2A中的并行处理器存储器222和/或如图1中的系统存储器104)中、要在一个或多个显示设备110A-110B上显示或用于由一个或多个处理器102或并行处理器112中的一个进一步处理的经处理的图形数据,这些栅格操作包括但不限于模版印制、z测试、混合等等。栅格操作单元526可被配置成用于压缩被写入到存储器的z数据或颜色数据并对从存储器读取的z数据或颜色数据解压缩。
机器学习概述
上文描述的体系结构可被应用以执行使用机器学习模型的训练和推断操作。机器学习在解决许多种类的任务方面已经是成功的。当训练和使用机器学习算法(例如,神经网络)时产生的计算本身自然地适合于高效的并行实现方式。相应地,诸如通用图形处理单元(GPGPU)之类的并行处理器已在深度神经网络的实际实现方式中扮演了重要角色。具有单指令多线程(SIMT)体系结构的并行图形处理器被设计成用于使图形管线中的并行处理的量最大化。在SIMT体系结构中,成组的并行线程尝试尽可能频繁地一起同步地执行程序指令以增加处理效率。由并行机器学习算法实现方式提供的效率允许使用高容量网络并且使得能够对较大的数据集训练那些网络。
机器学习算法是能够基于数据集合进行学习的算法。例如,机器学习算法可以被设计成用于对数据集内的高级抽象进行建模。例如,可以使用图像识别算法来确定给定的输入属于若干类别中的哪个类别;给定输入,回归算法可以输出数值;并且可以使用模式识别算法来生成经转换的文本或者执行文本到语音和/或语音识别。
示例性类型的机器学习算法是神经网络。存在许多类型的神经网络;简单类型的神经网络是前馈网络。前馈网络可以被实现为在其中按层来布置节点的非循环图。典型地,前馈网络拓扑包括由至少一个隐藏层分开的输入层和输出层。隐藏层将由输入层接收到的输入变换为对在输出层中生成输出有用的表示。网络节点经由边被完全连接到相邻层中的节点,但在每个层内的节点之间不存在边。在前馈网络的输入层的节点处接收到的数据经由激活函数被传播(即,“前馈”)至输出层的节点,该激活函数基于分别与连接这些层的边中的每一条边相关联的系数(“权重”)来计算网络中每个连续层的节点的状态。取决于正由正被执行的算法表示的特定模型,来自神经网络算法的输出可以采用各种形式。
在可使用机器学习算法对特定问题建模之前,使用训练数据集来训练算法。训练神经网络涉及:选择网络拓扑;使用表示正由网络建模的问题的训练数据的集合;以及调整权重,直到网络模型针对训练数据集的所有实例都以最小误差执行。例如,在针对神经网络的有监督学习训练过程期间,由网络响应于表示训练数据集中的实例的输入而产生的输出与该实例的“正确的”标记输出进行比较,计算表示输出与标记输出之间的差异的误差信号,并且随着误差信号通过网络的层被向后传播,调整与连接相关联的权重以使那个误差最小化。当根据训练数据集的实例生成的输出中的每个输出的误差被最小化时,网络被认为是“经训练的”。
机器学习算法的准确度会显著地受用于训练算法的数据集的质量影响。训练过程可能是计算密集型的,并且在常规通用处理器上可能需要大量的时间。相应地,使用并行处理硬件来训练许多类型的机器学习算法。这对于优化神经网络的训练是特别有用的,因为在调整神经网络中的系数时执行的计算本身自然地适于并行实现方式。具体地,许多机器学习算法和软件应用已被适配成利用通用图形处理设备内的并行处理硬件。
图6是机器学习软件栈600的广义图。机器学习应用602是能够被配置成用于使用训练数据集来训练神经网络或使用经训练的深度神经网络来实现机器智能的任何逻辑。机器学习应用602可包括用于神经网络的训练和推断功能和/或可用于在部署之前训练神经网络的专业软件。机器学习应用602可实现任何类型的机器智能,包括但不限于:图像识别、地图创建和定位、自主导航、语音合成、医学成像或语言翻译。示例机器学习应用602包括但不限于基于语音的虚拟助手、图像或面部识别算法、自主导航以及被用于训练由机器学习应用602使用的机器学习模型的软件工具。
可以经由机器学习框架604来启用用于机器学习应用602的硬件加速。机器学习框架604可提供机器学习基元的库。机器学习基元是通常由机器学习算法执行的基本操作。在没有机器学习框架604的情况下,将需要机器学习算法的开发者创建和优化与机器学习算法相关联的主计算逻辑,随后在开发新的并行处理器时重新优化计算逻辑。相反,机器学习应用可以被配置成使用由机器学习框架604提供的基元来执行必要的计算。示例性基元包括张量卷积、激活函数和池化,它们是在训练卷积神经网络(convolutional neuralnetwork,CNN)时执行的计算操作。机器学习框架604还可以提供基元以实现由许多机器学习算法执行的基本线性代数子程序,诸如,矩阵和向量操作。机器学习框架604的示例包括但不限于TensorFlow(张量流)、TensorRT、PyTorch、MXNet、Caffee以及其他高级机器学习框架。
机器学习框架604可处理从机器学习应用602接收到的输入数据,并生成至计算框架606的适当输入。计算框架606可抽象出提供给GPGPU驱动器608的底层指令,以使得机器学习框架604能够经由GPGPU硬件610来利用硬件加速而无需机器学习框架604非常熟悉GPGPU硬件610的体系结构。此外,计算框架606可以跨各种类型和世代的GPGPU硬件610来启用用于机器学习框架604的硬件加速。示例性计算框架606包括CUDA计算框架和相关联的机器学习库,诸如,CUDA深度神经网络(CUDA Deep Neural Network,cuDNN)库。机器学习软件栈600还可包括通信库或框架以促进多GPU和多节点计算。
GPGPU机器学习加速
图7图示通用图形处理单元700,其可以是图2A的并行处理器200或图1的(一个或多个)并行处理器112。通用处理单元(GPGPU)700可被配置成用于提供对由机器学习框架提供的基元的硬件加速的支持,以加速处理与训练深度神经网络相关联的类型的计算工作负载。此外,GPGPU 700可直接链接到GPGPU的其他实例以创建多GPU集群,从而改善尤其是深度神经网络的训练速度。基元也受支持以加速用于经部署的神经网络的推断操作。
GPGPU 700包括用于启用与主机处理器的连接的主机接口702。主机接口702可以是PCI快速接口。然而,主机接口还可以是供应方特定的通信接口或通信结构。GPGPU 700从主机处理器接收命令,并且使用全局调度器704以将与那些命令相关联的执行线程分发给处理集群706A-706H的集合。处理集群706A-706H共享缓存存储器708。缓存存储器708可充当用于处理集群706A-706H内的缓存存储器的较高级别的缓存。所图示的处理集群706A-706H可与如图2A中的处理集群214A-214N相对应。
GPGPU 700包括经由存储器控制器的集合712A-712B与处理集群706A-706H耦合的存储器714A-714B。存储器714A-714B可包括各种类型的存储器设备,包括动态随机存取存储器(DRAM)或图形随机存取存储器,诸如,同步图形随机存取存储器(SGRAM),包括图形双倍数据速率(GDDR)存储器。存储器714A-714B还可包括3D堆叠式存储器,包括但不限于高带宽存储器(HBM)。
处理集群706A-706H中的每一个可包括图形多处理器的集合,诸如,图2D的图形多处理器234、图3A的图形多处理器325、图3B的图形多处理器350或者可包括如图3C中的多核心组365A-365N。计算集群的图形多处理器包括能够以包括适用于机器学习计算的一系列精度来执行计算操作的多种类型的整数和浮点逻辑单元。例如,处理集群706A-706H中的每一个计算集群中的浮点单元的至少子集可被配置成用于执行16比特或32比特浮点操作,而浮点单元的不同子集可被配置成用于执行64比特浮点操作。
GPGPU 700的多个实例可以被配置成作为计算集群进行操作。由计算集群用于同步和数据交换的通信机制跨实施例而有所不同。例如,GPGPU 700的多个实例通过主机接口702进行通信。在一个实施例中,GPGPU 700包括I/O中枢709,该I/O中枢709将GPGPU 710与GPU链路710耦合,该GPU链路710启用至GPGPU的其他实例的直接连接。GPU链路710可耦合至专用的GPU至GPU桥接器,该GPU至GPU桥接器启用GPGPU 700的多个实例之间的通信和同步。任选地,GPU链路710与高速互连耦合,以将数据传送至其他GPGPU或并行处理器并且从其他GPGPU或并行处理器接收数据。GPGPU 700的多个实例可位于单独的数据处理系统中,并且可经由网络设备进行通信,该网络设备可经由主机接口702来访问。附加于或替代于主机接口702,GPU链路710可被配置成用于启用至主机处理器的连接。
尽管GPGPU 700的所图示的配置可被配置成用于训练神经网络,但是GPGPU 700的替代配置可被配置成用于在高性能或低功率推断平台内的部署。在推断配置中,相对于训练配置,GPGPU 700包括处理集群706A-706H中的更少的处理集群。此外,与存储器714A-714B相关联的存储器技术在推断配置与训练配置之间可有所不同。在一个实施例中,GPGPU700的推断配置可支持推断特定指令。例如,推断配置可提供对一个或多个8比特整数或浮点点积指令的支持,这一个或多个8比特整数或浮点点积指令通常在用于经部署的神经网络的推断操作期间被使用。
图8图示多GPU计算系统800。多GPU计算系统800可包括经由主机接口开关804被耦合至多个GPGPU 806A-806D的处理器802。主机接口开关804可以是将处理器802耦合至PCI快速总线的PCI快速开关设备,处理器802能够通过该PCI快速总线与GPGPU806A-806D的集合通信。多个GPGPU 806A-806D中的每一个可以是图7的GPGPU 700的实例。GPGPU 806A-806D可经由高速点到点GPU至GPU链路816的集合来互连。高速GPU至GPU链路可以经由专用GPU链路连接到GPGPU 806A-806D中的每一个,该专用GPU链路诸如图7中的GPU链路710。P2PGPU链路816启用GPGPU 806A-806D中的每一个GPGPU之间的直接通信,而不需要通过处理器802被连接到的主机接口总线进行的通信。利用定向到P2P GPU链路的GPU至GPU通信量,主机接口总线保持可用于系统存储器访问,或用于例如经由一个或多个网络设备与多GPU计算系统800的其他实例通信。虽然在图8中GPGPU 806A-806D经由主机接口开关804连接到处理器802,但是处理器802可替代地包括对P2P GPU链路816的直接支持,并直接连接到GPGPU806A-806D。在一个实施例中,P2P GPU链路816使多GPU计算系统800能够作为单个逻辑GPU进行操作。
机器学习神经网络实现方式
本文中描述的计算体系结构可被配置成用于执行特别适合于训练和部署用于机器学习的神经网络的一类并行处理。可以将神经网络概括为具有图关系的函数的网络。如本领域中所公知,存在在机器学习中使用的各种类型的神经网络实现方式。一种示例性类型的神经网络是如先前所描述的前馈网络。
第二示例性类型的神经网络是卷积神经网络(CNN)。CNN是用于处理具有已知的、栅格状拓扑的数据(诸如,图像数据)的专业的前馈神经网络。相应地,CNN通常用于计算机视觉和图像识别应用,但是它们也可用于其他类型的模式识别,诸如,语音和语言处理。CNN输入层中的节点被组织为“过滤器”的集合(由视网膜中发现的感受野激发的特征检测器),并且每个过滤器集合的输出被传播至网络的连续层中的节点。用于CNN的计算包括将卷积数学运算应用于每个过滤器以产生该过滤器的输出。卷积是由两个函数执行以产生第三函数的专业种类的数学运算,该第三函数是两个原始函数中的一个函数的修改版本。在卷积网络术语中,至卷积的第一函数可被称为输入,而第二函数可被称为卷积核。输出可被称为特征图。例如,至卷积层的输入可以是定义输入图像的各种颜色分量的多维数据数组。卷积核可以是多维参数数组,其中通过用于神经网络的训练过程来使参数适配。
循环神经网络(recurrent neural network,RNN)是包括层之间的反馈连接的一系列前馈神经网络。RNN通过跨神经网络的不同部分共享参数数据来启用对序列化数据进行建模。用于RNN的体系结构包括循环。这些循环表示变量的当前值在将来时刻对其自身值的影响,因为来自RNN的输出数据的至少部分被用作反馈以用于处理序列中的后续输入。由于语言数据可被组成的可变本质,这个特征使RNN变得对语言处理特别有用。
下文描述的附图呈现了示例性前馈网络、CNN网络和RNN网络,并且描述了用于分别训练和部署那些类型的网络中的每一者的一般过程。将理解,这些描述就本文中描述的任何特定实施例而言是示例性且是非限制性的,并且一般说来所图示的概念一般可应用于深度神经网络和机器学习技术。
上文描述的示例性神经网络可以用于执行深度学习。深度学习是使用深度神经网络的机器学习。与仅包括单个隐藏层的浅层神经网络不同,在深度学习中使用的深度神经网络是由多个隐藏层组成的人工神经网络。更深的神经网络通常训练起来是更加计算密集性的。然而,网络的附加的隐藏层启用多步模式识别,该多步模式识别得到相对于浅层机器学习技术的减小的输出误差。
在深度学习中使用的深度神经网络典型地包括用于执行特征识别的前端网络,该前端网络耦合至后端网络,该后端网络表示可基于提供给数学模型的特征表示来执行操作(例如,对象分类、语音识别等)的数学模型。深度学习使得机器学习能够在无需针对模型执行手工的特征工程化的情况下被执行。相反,深度神经网络可基于输入数据内的统计结构或相关性来学习特征。所学习的特征可被提供给数学模型,该数学模型可将所检测的特征映射至输出。由网络使用的数学模型通常专用于要执行的特定任务,并且不同的模型将用于执行不同的任务。
一旦将神经网络结构化,就可以将学习模型应用于网络以将网络训练成执行特定任务。学习模型描述如何调整模型内的权重以减小网络的输出误差。误差的反向传播是用于训练神经网络的常用方法。输入向量被呈现给网络以供处理。使用损失函数将网络的输出与期望输出进行比较,并且为输出层中的神经元中的每个神经元计算误差值。随后,向后传播误差值,直到每个神经元具有粗略地表示该神经元对原始输出的贡献的相关联的误差值。网络随后可使用算法(诸如,随机梯度下降算法)从那些误差中进行学习,以更新神经网络的权重。
图9A-图9B图示示例性卷积神经网络。图9A图示CNN内的各个层。如图9A中所示,用于对图像处理建模的示例性CNN可接收描述输入图像的红色、绿色、和蓝色(red,green,andblue,RGB)分量的输入902。输入902可由多个卷积层(例如,卷积层904、卷积层906)来处理。来自多个卷积层的输出任选地可由全连接层908的集合来处理。如先前针对前馈网络所描述,全连接层中的神经元具有至前一层中的所有激活的完全连接。来自全连接层908的输出可用于从网络生成输出结果。可使用矩阵乘法而不是卷积来计算全连接层908内的激活。并非所有CNN实现方式都利用全连接层908。例如,在一些实现方式中,卷积层906可生成CNN的输出。
卷积层被稀疏地连接,这与在全连接层908中发现的传统神经网络配置不同。传统神经网络层被完全连接,使得每个输出单元与每个输入单元相互作用。然而,如所图示,卷积层被稀疏地连接,因为感受野的卷积的输出(而不是感受野中的节点中的每个节点的相应状态值)被输入到后续层的节点。与卷积层相关联的核执行卷积操作,该卷积操作的输出被发送至下一层。在卷积层内执行的降维是使得CNN能够缩放以处理大图像的一个方面。
图9B图示CNN的卷积层内的示例性计算阶段。可以在卷积层914的三个阶段中处理至CNN的卷积层的输入912。这三个阶段可包括卷积阶段916、检测器阶段918和池化阶段920。卷积层914随后可将数据输出至连续的卷积层。网络的最终卷积层可生成输出特征图数据或提供至全连接层的输入,例如以生成用于至CNN的输入的分类值。
在卷积阶段916中并行地执行若干个卷积,以产生线性激活的集合。卷积阶段916可包括仿射变换,该仿射变换是可被指定为线性变换加平移的任何变换。仿射变换包括旋转、平移、缩放和这些变换的组合。卷积阶段计算连接到输入中的特定区域的函数(例如,神经元)的输出,这些特定区域可以被确定为与神经元相关联的局部区域。神经元计算神经元的权重与局部输入中神经元被连接到的区域的权重之间的点积。来自卷积阶段916的输出定义由卷积层914的连续阶段处理的线性激活的集合。
线性激活可由检测器阶段918处理。在检测器阶段918中,每个线性激活由非线性激活函数处理。非线性激活函数增加整体网络的非线性性质,而不影响卷积层的感受野。可使用若干种类型的非线性激活函数。一种特定类型是修正线性单元(rectified linearunit,ReLU),其使用被定义为f(x)=max(0,x)的激活函数,使得激活的阈值为零。
池化阶段920使用池化函数,该池化函数利用附近输出的摘要统计来替换第卷积层906的输出。池化函数可用于将平移不变性引入到神经网络中,使得至输入的小平移不改变经池化的输出。局部平移的不变性在其中特征在输入数据中的存在比特征的精确位置更重要的场景中可以是有用的。可以在池化阶段920期间使用各种类型的池化函数,包括最大池化、平均池化和l2范数池化。此外,一些CNN实现方式不包括池化阶段。相反,此类实现方式是相对于先前的卷积阶段具有增加的跨度的替代和附加卷积阶段。
来自卷积层914的输出随后可由下一层922处理。下一层922可以是附加的卷积层或是全连接层908中的一个层。例如,图9A的第一卷积层904可输出到第二卷积层906,而第二卷积层可输出到全连接层908中的第一层。
图10图示示例性循环神经网络1000。在循环神经网络(RNN)中,网络的先前状态影响网络的当前状态的输出。能以各种方式、使用各种函数来建立RNN。RNN的使用总体上围绕着使用数学模型以基于输入的先前序列来预测未来。例如,RNN可被用于执行统计语言建模,以在给定单词的先前序列的情况下预测即将到来的单词。所图示的RNN 1000可被描述为具有接收输入向量的输入层1002、用于实现循环函数的隐藏层1004、用于启用先前状态的‘记忆’的反馈机制1005以及用于输出结果的输出层1006。RNN 1000基于时间步长进行操作。处于给定时间步长的RNN的状态经由反馈机制1005、基于先前的时间步长而受影响。对于给定的时间步长,隐藏层1004的状态由先前状态以及当前时间步长处的输入来定义。处于第一时间步长的初始输入(x1)可由隐藏层1004处理。第二输入(x2)可由隐藏层1004使用在初始输入(x1)的处理期间确定的状态信息来处理。给定状态可被计算为st=f(Uxt+Wst-1),其中,U和W是参数矩阵。函数f总体上是非线性,诸如,双曲正切函数(Tanh)或修正函数f(x)=max(0,x)的变体。然而,在隐藏层1004中使用的特定的数学函数可取决于RNN1000的特定实现细节而有所不同。
除了所描述的基本CNN和RNN网络之外,还可启用针对那些网络的变体的加速。一个示例RNN变体是长短期记忆(long short term memory,LSTM)RNN。LSTM RNN能够学习长期依赖关系,该长期依赖关系可对于处理较长的语言序列而言是必需的。CNN的变体是卷积深度信念网络,该卷积深度信念网络具有与CNN类似的结构,并且以与深度信念网络类似的方式被训练。深度信念网络(deep belief network,DBN)是由随机性(随机)变量的多个层组成的生成式神经网络。DBN可使用贪婪无监督学习来逐层训练。DBN的所学习的权重随后可被用来通过确定用于神经网络的权重的最佳初始集合来提供预训练神经网络。在进一步的实施例中,启用用于强化学习的加速。在强化学习中,人工代理通过与其环境交互来学习。该代理被配置成用于优化某些目标以使累积回报最大化。
图11图示深度神经网络的训练和部署。一旦已针对任务将给定的网络结构化,就使用训练数据集1102来训练神经网络。已开发各种训练框架1104来启用训练过程的硬件加速。例如,图6的机器学习框架604可被配置为训练框架1104。训练框架1104可接入未经训练的神经网络1106,并且使得未经训练的神经网络能够使用本文描述的并行处理资源被训练以生成经训练的神经网络1108。
为了开始训练过程,可随机地选择权重或通过使用深度信念网络进行预训练来选择初始权重。随后,以有监督或无监督方式执行训练循环。
有监督学习是在其中训练被执行为介导操作的学习方法,诸如,当训练数据集1102包括与输入的期望输出配对的该输入时,或在训练数据集包括具有已知输出的输入并且神经网络的输出被手动分级的情况下。网络处理输入,并且将所得的输出与预期输出或期望输出的集合进行比较。随后,通过系统往回传播误差。训练框架1104可进行调整以调整控制未经训练的神经网络1106的权重。训练框架1104可提供工具以监测未经训练的神经网络1106正在多好地收敛于适合基于已知的输入数据生成正确答案的模型。随着网络的权重被调整以细化由神经网络生成的输出,训练过程反复地发生。训练过程可继续,直到神经网络达到与经训练的神经网络1108相关联的统计上期望的准确度。经训练的神经网络1108随后可被部署为实现任何数量的机器学习操作,以基于新数据1112的输入来生成推断结果1114。
无监督学习是在其中网络尝试使用未标记数据来训练其自身的学习方法。因此,针对无监督学习,训练数据集1102将包括不具有任何相关联的输出数据的输入数据。未经训练的神经网络1106可学习未标记输入内的分组,并且可确定单独的输入如何与整个数据集相关。无监督训练可被用于生成自组织图,该自组织图是能够执行在降低数据的维度方面有用的操作的一类经训练的神经网络1108。无监督训练还可被用于执行异常检测,该异常检测允许标识输入数据集中的、从数据的正常模式偏离的数据点。
还可采用有监督训练和无监督训练的变体。半监督学习是在其中在训练数据集1102中包括具有相同分布的经标记数据和未标记数据的混合的技术。渐进式学习是有监督学习的变体,其中连续地使用输入数据以进一步训练模型。渐进式学习使得经训练的神经网络1108能够适配于新数据1112,而不忘记在初始训练期间根植在网络内的知识。
无论是有监督还是无监督的,用于特别深的神经网络的训练过程对于单个计算节点可能是过于计算密集的。可以使用计算节点的分布式网络而不是使用单个计算节点来加速训练过程。
图12A是图示分布式学习的框图。分布式学习是使用多个分布式计算节点来执行神经网络的有监督或无监督训练的训练模型。分布式计算节点各自可包括一个或多个主机处理器或通用处理节点中的一个或多个通用处理节点,通用处理节点诸如图7中的高度并行的通用图形处理单元700。如所图示,能以模型并行性1202、数据并行性1204或模型和数据并行性的组合1206来执行分布式学习。
在模型并行性1202中,分布式系统中的不同计算节点可为单个网络的不同部分执行训练计算。例如,神经网络的每一层可由分布式系统的不同处理节点来训练。模型并行性的益处包括缩放到特别大的模型的能力。分割与神经网络的不同层相关联的计算使得能够训练非常大型的神经网络,在非常大型的神经网络中,所有层的权重将无法适配到单个节点的存储器中。在一些实例中,模型并行性在执行大型神经网络的无监督训练时可以是特别有用的。
在数据并行性1204中,分布式网络的不同节点具有模型的完整实例,并且每个节点接收数据的不同部分。来自不同节点的结果随后被组合。虽然实现数据并行性的不同方式是可能的,但是数据并行性训练方式全都需要将结果组合并使每个节点之间的模型参数同步的技术。用于组合数据的示例性方式包括参数求平均以及基于更新的数据并行性。参数求平均对训练数据的子集训练每个节点,并且将全局参数(例如,权重、偏置)设置为来自每个节点的参数的平均值。参数求平均使用维护参数数据的中央参数服务器。基于更新的数据并行性与参数求平均类似,例外在于,对模型的更新被传输,而不是将参数从节点传输到参数服务器。另外,能以分散化的方式执行基于更新的数据并行性,其中更新被压缩并且在节点之间传输。
可以例如在其中每个计算节点包括多个GPU的分布式系统中实现组合式模型和数据并行性1206。每个节点可具有模型的完整实例,其中每个节点内的单独GPU被用于训练模型的不同部分。
分布式训练相对于在单个机器上的训练已增加开销。然而,本文中描述的并行处理器和GPGPU各自可实现各种技术以降低分布式训练的开销,包括用于启用高带宽GPU至GPU数据传输和经加速的远程数据同步的技术。
图12B是图示可编程网络接口1210和数据处理单元的框图。可编程网络接口1210是可用于加速分布式环境内的基于网络的计算任务的可编程网络引擎。可编程网络接口1210可经由主机接口1270与主机系统耦合。可编程网络接口1210可用于加速用于主机系统的CPU或GPU的网络或存储操作。主机系统例如可以是用于执行分布式训练的分布式学习系统的节点,例如,如图12A中所示。主机系统也可以是数据中心内的数据中心节点。
在一个实施例中,可由可编程网络接口1210加速对包含模型数据的远程存储装置的存取。例如,可编程网络接口1210可被配置成用于将远程存储设备呈现为主机系统的本地存储设备。可编程网络接口1210还可加速在主机系统的GPU与远程系统的GPU之间执行的远程直接存储器存取(RDMA)操作。在一个实施例中,可编程网络接口1210可启用存储功能,诸如但不限于NVME-oF。可编程网络接口1210还可代表主机系统加速用于远程存储装置的加密、数据完整性、压缩和其他操作,从而允许远程存储装置逼近直接附连至主机系统的存储设备的等待时间。
可编程网络接口1210还可代表主机系统执行资源分配和管理。存储安全操作可被迁移到可编程网络接口1210,并且与远程存储资源的分配和管理协同地执行。用于管理对远程存储装置的存取的、否则将由主机系统的处理器执行的基于网络的操作可替代地由可编程网络接口1210执行。
在一个实施例中,网络和/或数据安全操作可从主机系统迁移到可编程网络接口1210。用于数据中心节点的数据中心安全策略可由可编程网络接口1210而不是主机系统的处理器来处置。例如,可编程网络接口1210可检测并缓解在主机系统上的被尝试的基于网络的攻击(例如,DDoS),从而防止攻击损害主机系统的可用性。
可编程网络接口1210可包括经由多个处理器核心1222执行操作系统的片上系统(SoC 1220)。处理器核心1222可包括通用处理器(例如,CPU)核心。在一个实施例中,处理器核心1222还可包括一个或多个GPU核心。SoC 1220可执行存储在存储器设备1240中的指令。存储设备1250可存储本地操作系统数据。存储设备1250和存储器设备1240还可用于对用于主机系统的远程数据进行缓存。网络端口1260A-1260B启用至网络或结构的连接,并且促进针对SoC 1220的网络访问,并经由主机接口1270促进针对主机系统的网络访问。可编程网络接口1210还可包括I/O接口1275,诸如,USB接口。I/O接口1275可用于将外部设备耦合至可编程网络接口1210或耦合为调试接口。可编程网络接口1210还包括管理接口1230,该管理接口1230使主机设备上的软件能够管理和配置可编程网络接口1210和/或SoC 1220。在一个实施例中,可编程网络接口1210还可包括一个或多个加速器或GPU 1245以接受并行计算任务从经由网络端口1260A-1260B耦合的SoC 1220、主机系统或远程系统的迁移。
示例性机器学习应用
可以应用机器学习以解决各种技术问题,包括但不限于计算机视觉、自主驾驶和导航、语音识别以及语言处理。计算机视觉在传统上已是机器学习应用的最活跃研究领域之一。计算机视觉的应用范围为从重现人类视觉能力(诸如,识别人脸)到创建新类别的视觉能力。例如,计算机视觉应用可以配置成用于从在视频中可见的物体中诱发的振动来识别声波。并行处理器加速的机器学习使得计算机视觉应用能够使用比先前可行的训练数据集显著更大的训练数据集来训练,并且使得推断系统能够使用低功率并行处理器来部署。
并行处理器加速的机器学习具有自主驾驶应用,包括车道和道路标志识别、障碍物规避、导航和驾驶控制。经加速的机器学习技术可用于基于定义对特定训练输入的适当响应的数据集来训练驾驶模型。本文中描述的并行处理器能够实现对用于自主驾驶解决方案的日益复杂的神经网络的快速训练,并且能够实现将低功率推断处理器部署在适合于集成到自主交通工具中的移动平台中。
并行处理器加速的深度神经网络已启用用于自动语音识别(automatic speechrecognition,ASR)的机器学习方式。ASR包括创建在给定的输入声序列的情况下计算最有可能的语言序列的函数。使用深度神经网络的经加速的机器学习已启用对于先前用于ASR的隐马尔可夫模型(hidden Markov model,HMM)和高斯混合模型(Gaussian mixturemodel,GMM)的替代。
并行处理器加速的机器学习还可以用于加速自然语言处理。自动学习过程可以利用统计推断算法以产生对于有误差的或不熟悉的输入稳健的模型。示例性自然语言处理器应用包括人类语言之间的自动机器翻译。
可以将用于机器学习的并行处理平台划分为训练平台和部署平台。训练平台总体上是高度并行的,并且包括用于加速多GPU单节点训练和多节点多GPU训练的优化。适用于训练的示例性并行处理器包括图7的通用图形处理单元700和图8的多GPU计算系统800。相反,经部署的机器学习平台一般包括适于在诸如相机、自主机器人和自主交通工具之类的产品中使用的较低功率的并行处理器。
此外,还可应用机器学习技术来加速或增强图形处理活动。例如,可将机器学习模型训练成用于识别由GPU加速的应用生成的输出并生成该输出的放大版本。可应用此类技术以加速生成用于游戏应用的高分辨率图像。各种其他图形管线活动可受益于机器学习的使用。例如,可将机器学习模型训练成用于对几何数据执行曲面细分操作以增加几何模型的复杂度,从而允许从具有相对较低的细节的几何体自动地生成细节更精细的几何体。
图13图示适于使用经训练的模型执行推断的示例性推断片上系统(SOC)1300。SOC1300可集成处理部件,包括媒体处理器1302、视觉处理器1304、GPGPU 1306和多核心处理器1308。GPGPU 1306可以是本文所描述的GPGPU,诸如,GPGPU 700,并且多核心处理器1308可以是本文中描述的多核心处理器,诸如,多核心处理器405-406。SOC 1300可附加地包括片上存储器1305,该片上存储器1305可启用能够由处理部件中的每个处理部件访问的共享片上数据池。处理部件可针对低功率操作进行优化,以启用对包括自主交通工具和自主机器人的各种机器学习平台的部署。例如,SOC 1300的一个实现方式可被用作用于自主交通工具的主控制系统的部分。在SOC 1300被配置成在自主交通工具中使用的情况下,SOC被设计并被配置成用于符合部署管辖的相关功能安全性标准。
在操作期间,媒体处理器1302和视觉处理器1304可协同地工作以加速计算机视觉操作。媒体处理器1302可启用对多个高分辨率(例如,4K、8K)视频流的低等待时间解码。经解码的视频流可被写入片上存储器1305中的缓冲器。视觉处理器1304随后可解析经解码的视频,并且对经解码的视频的帧执行初步处理操作,以准备好使用经训练的图像识别模型来处理帧。例如,视觉处理器1304可加速用于CNN的卷积操作,该CNN用于对高分辨率视频数据执行图像识别,同时后端模型计算由GPGPU1306执行。
多核心处理器1308可包括控制逻辑,该控制逻辑用于辅助对由媒体处理器1302和视觉处理器1304执行的数据传输和共享存储器操作的定序和同步。多核心处理器1308还可充当应用处理器,以用于执行能够利用GPGPU 1306的推断计算能力的软件应用。例如,导航和驾驶逻辑的至少部分可在多核心处理器1308上执行的软件中实现。此类软件可直接将计算工作负载发出至GPGPU 1306,或者计算工作负载可被发出至多核心处理器1308,该多核心处理器1308可将那些操作的至少部分迁移至GPGPU 1306。
GPGPU 1306可包括计算集群,诸如,通用图形处理单元700内的低功率配置的处理集群706A-706H。GPGPU 1306内的计算集群可支持专门优化成在经训练的神经网络上执行推断计算的指令。例如,GPGPU 1306可支持用于执行诸如8比特和4比特整数向量操作之类的低精度计算的指令。
附加系统概述
图14是处理系统1400的框图。图14的具有与本文中任何其他附图的元件相同或类似名称的元件描述与其他附图中相同的元件,能以与其他附图中类似的方式进行操作或运行,可包括相同的部件,并且可链接到其他实体,该实体如本文中其他地方所描述的那些实体,但不限于此。系统1400可在以下各项中被使用:单处理器桌面型电脑系统、多处理器工作站系统或具有大量处理器1402或处理器核心1407的服务器系统。系统1400可以是被并入在片上系统(SoC)集成电路内的处理平台,该片上系统(SoC)集成电路用于在移动设备、手持式设备或嵌入式设备中使用,诸如,用于在具有至局域网或广域网的有线或无线连接性的物联网(Internet-of-things,IoT)设备内使用。
系统1400可以是具有与图1的那些部件对应的部件的处理系统。例如,在不同配置中,(一个或多个)处理器1402或(一个或多个)处理器核心1407可与图1的(一个或多个)处理器102相对应。(一个或多个)图形处理器1408可与图1的(一个或多个)并行处理器112相对应。外部图形处理器1418可以是图1的(一个或多个)插入式设备120中的一个。
系统1400可包括以下各项,可与以下各项耦合,或可集成在以下各项内:基于服务器的游戏平台;游戏控制台,包括游戏和媒体控制台;移动游戏控制台、手持式游戏控制台或在线游戏控制台。系统1400可以是移动电话、智能电话、平板计算设备或移动互联网连接的设备(诸如,具有低内部存储容量的膝上型电脑)的部分。处理系统1400也可包括以下各项,与以下各项耦合,或被集成在以下各项内:可穿戴设备,诸如,智能手表可穿戴设备;智能眼镜或服装,其利用增强现实(augmented reality,AR)或虚拟现实(virtual reality,VR)特征来增强,以提供视觉、音频或触觉输出来补充现实世界视觉、音频或触觉体验或以其他方式提供文本、音频、图形、视频、全息图像或视频、或触觉反馈;其他增强现实(AR)设备;或其他虚拟现实(VR)设备。处理系统1400可包括电视机或机顶盒设备,或可以是电视机或机顶盒设备的部分。系统1400可包括自动驾驶交通工具,与自动驾驶交通工具耦合,或集成在自动驾驶交通工具内,该自动驾驶交通工具诸如,公共汽车、拖拉机拖车、汽车、电机或电力循环、飞机或滑翔机(或其任何组合)。自动驾驶交通工具可使用系统1400来处理在该交通工具周围感测到的环境。
一个或多个处理器1402可包括一个或多个处理器核心1407,该一个或多个处理器核心1407用于处理指令,这些指令当被执行时,执行用于系统和用户软件的操作。一个或多个处理器核心1407中的至少一个处理器核心可被配置成用于处理特定的指令集1409。指令集1409可促进复杂指令集计算(Complex Instruction Set Computing,CISC)、精简指令集计算(Reduced Instruction Set Computing,RISC)或经由超长指令字(Very LongInstruction Word,VLIW)的计算。一个或多个处理器核心1407可以处理不同的指令集1409,不同的指令集1409可包括用于促进对其他指令集的仿真的指令。处理器核心1407还可包括其他处理设备,诸如,数字信号处理器(Digital Signal Processor,DSP)。
处理器1402可包括缓存存储器1404。取决于体系结构,处理器1402可具有单个内部缓存或多级的内部缓存。在一些实施例中,缓存存储器在处理器1402的各种部件之间被共享。在一些实施例中,处理器1402也使用外部缓存(例如,第三级(L3)缓存或最后一级缓存(Last LevelCache,LLC))(未示出),可使用已知的缓存一致性技术在处理器核心1407之间共享该外部缓存。寄存器堆1406可附加地被包括在处理器1402中,并且可包括用于存储不同类型的数据的不同类型的寄存器(例如,整数寄存器、浮点寄存器、状态寄存器以及指令指针寄存器)。一些寄存器可以是通用寄存器,而其他寄存器可以专用于处理器1402的设计。
一个或多个处理器1402可与一个或多个接口总线1410耦合,以在处理器1402与系统1400中的其他部件之间传送通信信号,诸如,地址、数据、或控制信号。在这些实施例中的一个实施例中,接口总线1410可以是处理器总线,诸如,直接媒体接口(Direct MediaInterface,DMI)总线的某个版本。然而,处理器总线不限于DMI总线,并且可包括一个或多个外围部件互连总线(例如,PCI、PCI快速)、存储器总线或其他类型的接口总线。例如,(一个或多个)处理器1402可包括集成存储器控制器1416和平台控制器中枢1430。存储器控制器1416促进存储器设备与系统1400的其他部件之间的通信,而平台控制器中枢(platformcontroller hub,PCH)1430提供经由本地I/O总线至I/O设备的连接。
存储器设备1420可以是动态随机存取存储器(DRAM)设备、静态随机存取存储器(static random-access memory,SRAM)设备、闪存设备、相变存储器设备或具有合适的性能以充当进程存储器的某个其他存储器设备。存储器设备1420可以例如作为用于系统1400的系统存储器来操作,以存储数据1422和指令1421,用于在一个或多个处理器1402执行应用或进程时使用。存储器控制器1416也与任选的外部图形处理器1418耦合,该任选的外部图形处理器1418可与处理器1402中的一个或多个图形处理器1408通信以执行图形操作和媒体操作。在一些实施例中,可由加速器1412辅助图形操作、媒体操作和/或计算操作,该加速器1412是可被配置成用于执行专业的图形操作、媒体操作或计算操作的集合的协处理器。例如,加速器1412可以是用于优化机器学习或计算操作的矩阵乘法加速器。加速器1412可以是光线追踪加速器,该光线追踪加速器可用于与图形处理器1408协同地执行光线追踪操作。在一个实施例中,可替代加速器1412使用外部加速器1419,或可与加速器1412协同地使用外部加速器1419。
可提供显示设备1411,该显示设备1411可连接到(一个或多个)处理器1402。显示设备1411可以是以下各项中的一项或多项:内部显示设备,如在移动电子设备或膝上型电脑设备中;或经由显示接口(例如,显示端口等)附接的外部显示设备。显示设备1411可以是头戴式显示器(head mounted display,HMD),诸如,用于在虚拟现实(VR)应用或增强现实(AR)应用中使用的立体显示设备。
平台控制器中枢1430可以使外围设备能够经由高速I/O总线连接到存储器设备1420和处理器1402。I/O外围设备包括但不限于音频控制器1446、网络控制器1434、固件接口1428、无线收发器1426、触摸传感器1425、数据存储设备1424(例如,非易失性存储器、易失性存器、硬盘驱动器、闪存、NAND、3D NAND、3D Xpoint/Optane等)。数据存储设备1424可以经由存储接口(例如,SATA)或经由外围总线(诸如,外围部件互连总线(例如,PCI、PCI快速))连接。触摸传感器1425可以包括触摸屏传感器、压力传感器或指纹传感器。无线收发器1426可以是Wi-Fi收发器、蓝牙收发器或移动网络收发器,该移动网络收发器诸如3G、4G、5G或长期演进(Long-Term Evolution,LTE)收发器。固件接口1428启用与系统固件的通信,并且可以例如是统一可扩展固件接口(unified extensible firmware interface,UEFI)。网络控制器1434可启用至有线网络的网络连接。在一些实施例中,高性能网络控制器(未示出)与接口总线1410耦合。音频控制器1446可以是多声道高清音频控制器。在这些实施例中的一些实施例中,系统1400包括用于将传统(例如,个人系统2(Personal System 2,PS/2))设备耦合至系统的任选的传统I/O控制器1440。平台控制器中枢1430还可以连接到一个或多个通用串行总线(Universal Serial Bus,USB)控制器1442,以连接到输入设备,诸如,键盘和鼠标1443组合、相机1444或其他USB输入设备。
将领会,所示的系统1400是示例性而非限制性的,因为也可以使用以不同方式配置的其他类型的数据处理系统。例如,存储器控制器1416和平台控制器中枢1430的实例可以集成到分立的外部图形处理器中,该分立的外部图形处理器诸如外部图形处理器1418。平台控制器中枢1430和/或存储器控制器1416可以在一个或多个处理器1402外部。例如,系统1400可包括外部存储器控制器1416和平台控制器中枢1430,该外部存储器控制器1416和平台控制器中枢1430可以被配置为在与(一个或多个)处理器1402通信的系统芯片组内的存储器控制器中枢和外围控制器中枢。
例如,可使用电路板(“橇板(sled)”),部件(诸如,CPU、存储器和其他部件)被放置在该电路板上,并且在该电路板上部件(诸如,CPU、存储器和其他部件)经设计以实现提升的热性能。诸如处理器之类的处理部件可位于橇板的顶侧上,而诸如DIMM之类的附近存储器位于橇板的底侧上。作为由该设计提供的增强的气流的结果,部件能以比在典型系统中更高的频率和功率等级来操作,由此提高性能。此外,橇板被配置成用于盲配机架中的功率和数据通信线缆,由此增强它们被快速地移除、升级、重新安装和/或替换的能力。类似地,位于橇板上的各个部件(诸如,处理器、加速器、存储器和数据存储驱动器)由于它们距彼此的增加的间距而被配置成易于升级。在说明性实施例中,部件附加地包括用于证明它们的真实性的硬件认证特征。
数据中心可利用支持多个其他网络体系结构的单个网络体系结构(“结构”),多个其他网络体系结构包括以太网和全方位路径。橇板可经由光纤耦合至交换机,这提供比典型的双绞线布线(例如,5类、5e类、6类等)更高的带宽和更低的等待时间。由于高带宽、低等待时间的互连和网络体系结构,数据中心在使用中可集中在物理上分散的诸如存储器、加速器(例如,GPU、图形加速器、FPGA、ASIC、神经网络和/或人工智能加速器等)和数据存储驱动器之类的资源,并且根据需要将它们提供给计算资源(例如,处理器),从而使计算资源能够就好像被集中的资源在本地那样访问这些被集中的资源。
功率供应或功率源可将电压和/或电流提供给系统1400或本文中描述的任何部件或系统。在一个示例中,功率供应包括用于插入到墙壁插座中的AC到DC(交流到直流)适配器。此类AC功率可以是可再生能源(例如,太阳能)功率源。在一个示例中,功率源包括DC功率源,诸如,外部AC到DC转换器。功率源或功率供应还可包括用于通过接近充电场来充电的无线充电硬件。功率源可包括内部电池、交流供应、基于动作的功率供应、太阳能功率供应或燃料电池源。
图15A-图15C图示计算系统和图形处理器。图15A-图15C的具有与本文中任何其他附图的元件相同或类似名称的元件描述与其他附图中相同的元件,能以与其他附图中类似的方式进行操作或运行,可包括相同的部件,并且可链接到其他实体,该实体如本文中其他地方所描述的那些实体,但不限于此。
图15A是处理器1500的框图,该处理器1500可以是处理器1402中的一个处理器的变体,并且可以替代那些处理器中的一个处理器被使用。因此,本文中结合处理器1500对任何特征的公开也公开了对应的与(一个或多个)处理器1402的结合,但不限于此。处理器1500可具有一个或多个处理器核心1502A-1502N、集成存储器控制器1514以及集成图形处理器1508。在集成图形处理器1508被排除的情况下,包括处理器的系统将包括系统芯片组内的、或经由系统总线耦合的图形处理器设备。处理器1500可包括附加的核心,这些附加的核心最多为由虚线框表示的附加核心1502N并包括由虚线框表示的附加核心1502N。处理器核心1502A-1502N中的每一个包括一个或多个内部缓存单元1504A-1504N。在一些实施例中,每个处理器核心1502A-1502N也具有对一个或多个共享缓存单元1506的访问权。内部缓存单元1504A-1504N和共享缓存单元1506表示处理器1500内的缓存存储器层次体系。缓存存储器层次体系可包括每个处理器核心内的至少一个级别的指令和数据缓存以及一个或多个级别的共享的中级缓存,诸如,第二级(L2)、第三级(L3)、第四级(L4)或其他级别的缓存,其中,在外部存储器之前的最高级别的缓存被分类为LLC。在一些实施例中,缓存一致性逻辑维持各缓存单元1506与1504A-1504N之间的一致性。
处理器1500还可包括一个或多个总线控制器单元的集合1516和系统代理核心1510。一个或多个总线控制器单元1516管理外围总线的集合,诸如,一个或多个PCI总线或PCI快速总线。系统代理核心1510提供对各处理器部件的管理功能。系统代理核心1510可包括用于管理对各种外部存储器设备(未示出)的访问的一个或多个集成存储器控制器1514。
例如,处理器核心1502A-1502N中的一个或多个处理器核心可包括针对同步多线程操作的支持。系统代理核心1510包括用于在多线程处理期间协调并操作核心1502A-1502N的部件。系统代理核心1510可附加地包括功率控制单元(power control unit,PCU),该功率控制单元(PCU)包括用于调节处理器核心1502A-1502N和图形处理器1508的功率状态的逻辑和部件。
处理器1500可附加地包括用于执行图形处理操作的图形处理器1508。在这些实施例中的一些实施例中,图形处理器1508与共享缓存单元的集合1506以及系统代理核心1510耦合,该系统代理核心1510包括一个或多个集成存储器控制器1514。系统代理核心1510还可包括用于将图形处理器输出驱动到一个或多个经耦合的显示器的显示控制器1511。显示控制器1511还可以是经由至少一个互连与图形处理器耦合的单独模块,或者可以集成在图形处理器1508内。
基于环的互连1512可用于耦合处理器1500的内部部件。然而,可以使用替代的互连单元,诸如,点到点互连、交换式互连或其他技术,包括本领域中公知的技术。在具有基于环的互连1512的这些实施例中的一些实施例中,图形处理器1508经由I/O链路1513与基于环的互连1512耦合。
示例性I/O链路1513表示多个各种各样的I/O互连中的至少一种,包括促进各处理器部件与高性能存储器模块1518(诸如,eDRAM模块或高带宽存储器(high-bandwidthmemory,HMB)模块)之间的通信的封装上I/O互连。任选地,处理器核心1502A-1502N中的每一个以及图形处理器1508可将高性能存储器模块1518用作共享的最后一级缓存。
处理器核心1502A-1502N可以例如是执行相同的指令集体系结构的同构核心。替代地,处理器核心1502A-1502N在指令集体系结构(instruction set architecture,ISA)方面是异构的,其中,处理器核心1502A-1502N中的一个或多个执行第一指令集,而其他核心中的至少一个执行第一指令集的子集或不同的指令集。处理器核心1502A-1502N在微体系结构方面可以是异构的,其中,具有相对较高功耗的一个或多个核心与具有较低功耗的一个或多个功率核心耦合。作为另一示例,处理器核心1502A-1502N在计算能力方面是异构的。此外,处理器1500可在一个或多个芯片上实现,或者被实现为除其他部件之外还具有所图示的部件的SoC集成电路。
图15B是根据本文中描述的一些实施例的图形处理器核心块1519的硬件逻辑的框图。在一些实施例中,图15B的具有与本文中任何其他附图的元件相同的附图标记(或名称)的元件能以与本文中其他地方所描述的方式类似的方式进行操作或运行。在一个实施例中,图形处理器核心块1519是图形处理器的一个分区的示例。图形处理器核心块1519可以被包括在图15A的集成图形处理器1508或分立的图形处理器、并行处理器和/或计算加速器内。如本文中所描述的图形处理器可包括基于目标功率和性能包络的多个图形核心块。每个图形处理器核心块1519可包括与多个图形核心1521A-1521F耦合的功能块1530,多个图形核心1521A-1521F包括固定功能逻辑和通用可编程逻辑的模块化的块。图形处理器核心块1519还包括能够由所有的图形核心1521A-1521F、栅格化器逻辑1537和附加的固定功能逻辑1538访问的共享/缓存存储器1536。
在一些实施例中,功能块1530包括可以由图形处理器核心块1519中的所有图形核心共享的几何/固定功能管线1531。在各实施例中,几何/固定功能管线1531包括3D几何管线、视频前端单元、线程生成器和全局线程调遣器以及统一返回缓冲器管理器,该统一返回缓冲器管理器管理统一返回缓冲器。在一个实施例中,功能块1530还包括图形SoC接口1532、图形微控制器1533和媒体管线1534。图形SoC接口1532提供图形处理器核心块1519与图形处理器或计算加速器SoC内的其他核心块之间的接口。图形微控制器1533是可被配置成用于管理图形处理器核心块1519的各种功能的可编程子处理器,这些功能包括线程调遣、调度和抢占。媒体管线1534包括用于促进多媒体数据(包括图像和视频数据)的解码、编码、预处理和/或后处理的逻辑。媒体管线1534经由对图形核心1521A-1521F内的计算或采样逻辑的请求来实现媒体操作。一个或多个像素后端1535也可被包括在功能块1530内。像素后端1535包括用于存储像素颜色值的缓冲存储器,并且能够执行对经渲染的像素数据的混合操作和无损颜色压缩。
在一个实施例中,图形SoC接口1532使图形处理器核心块1519能够与SoC内的或经由外围接口与SoC耦合的系统主机CPU内的通用应用处理器核心(例如,CPU)和/或其他部件进行通信。图形SoC接口1532还启用与片外存储器层次体系元件的通信,片外存储器层次体系元件诸如,共享的最后一级缓存存储器、系统RAM和/或嵌入式片上或封装上DRAM。SoC接口1532还能够启用与SoC内的诸如相机成像管线之类的固定功能设备的通信,并且启用全局存储器原子性的使用和/或实现全局存储器原子性,该全局存储器原子性可在图形处理器核心块1519与SoC内的CPU之间被共享。图形SoC接口1532还可实现针对图形处理器核心块1519的功率管理控制,并且启用图形处理器核心块1519的时钟域与SoC内的其他时钟域之间的接口。在一个实施例中,图形SoC接口1532使得能够从命令流转化器和全局线程调遣器接收命令缓冲器,该命令流转化器和全局线程调遣器被配置成用于将命令和指令提供给图形处理器内的一个或多个图形核心中的每一个图形核心。命令和指令在媒体操作要被执行时能够被调遣到媒体管线1534,并且在图形处理操作要被执行时能够被调遣到几何和固定功能管线1531。当计算操作要被执行时,计算调遣逻辑能够将命令调遣到图形核心1521A-1521F,从而绕过几何管线和媒体管线。
图形微控制器1533可被配置成用于执行针对图形处理器核心块1519的各种调度任务和管理任务。在一个实施例中,图形微控制器1533可以执行在图形核心1521A-1521F内的各向量引擎1522A-1522F、1524A-1524F和矩阵引擎1523A-1523F、1525A-1525F上调度的图形工作负载和/或计算工作负载。在该调度模型中,在包括图形处理器核心块1519的SoC的CPU核心上执行的主机软件可将工作负载提交到多个图形处理器门铃(doorbell)中的一个图形处理器门铃,这调用了对适当的图形引擎的调度操作。调度操作包括:确定接下来要运行哪个工作负载,将工作负载提交到命令流转化器,抢占在引擎上运行的现有工作负载,监测工作负载的进度,以及当工作负载完成时通知主机软件。在一个实施例中,图形微控制器1533还能够促进图形处理器核心块1519的低功率或空闲状态,从而向图形处理器核心块1519提供独立于操作系统和/或系统上的图形驱动器软件跨低功率状态转变来保存和恢复图形处理器核心块1519内的寄存器的能力。
图形处理器核心块1519可具有多于或少于所图示的图形核心1521A-1521F,最多N个模块化图形核心。对于每个具有N个图形核心的集合,图形处理器核心块1519还可包括:共享/缓存存储器1536,其可被配置为共享存储器或缓存存储器;栅格化器逻辑1537;以及附加的固定功能逻辑1538,用于加速各种图形和计算处理操作。
在每个图形核心1521A-1521F内包括可用于响应于由图形管线、媒体管线或着色器程序作出的请求而执行图形操作、媒体操作和计算操作的执行资源的集合。图形核心1521A-1521F包括多个向量引擎1522A-1522F、1524A-1524F、矩阵加速单元1523A-1523F、1525A-1525D、缓存/共享本地存储器(shared local memory,SLM)、采样器1526A-1526F以及光线追踪单元1527A-1527F。
向量引擎1522A-1522F、1524A-1524F是能够执行浮点和整数/定点逻辑操作以服务于图形操作、媒体操作或计算操作(包括图形程序、媒体程序或计算/GPGPU程序)的通用图形处理单元。向量引擎1522A-1522F、1524A-1524F能够使用SIMD执行模式、SIMT执行模式或SIMT+SIMD执行模式、以可变向量宽度进行操作。矩阵加速单元1523A-1523F、1525A-1525D包括矩阵-矩阵和矩阵-向量加速逻辑,该矩阵-矩阵和矩阵-向量加速逻辑改善矩阵操作、尤其是用于机器学习的低精度和混合精度(例如,INT8、FP16、BF16、FP8)矩阵操作的性能。在一个实施例中,矩阵加速单元1523A-1523F、1525A-1525D中的每一个包括能够对矩阵元素执行并发的矩阵乘法或点积操作的处理元件的一个或多个脉动阵列。
采样器1526A-1526F能够将媒体数据或纹理数据读入存储器,并且能够基于经配置的采样器状态和正被读取的纹理/媒体格式以不同方式对数据采样。在向量引擎1522A-1522F、1524A-1524F或矩阵加速单元1523A-1523F、1525A-1525D上执行的线程能够利用图形核心1521A-1521F中的每个图形核心内的缓存/SLM 1528A-1528F。缓存/SLM 1528A-1528F能够被配置为在相应的图形核心1521A-1521F中的每个图形核心的本地的缓存存储器或共享存储器的池。图形核心1521A-1521F内的光线追踪单元1527A-1527F包括光线遍历/相交电路,该光线遍历/相交电路用于使用包围体层次体系(BVH)来执行光线遍历并标识封围在BVH体积内的光线与基元之间的相交。在一个实施例中,光线追踪单元1527A-1527F包括用于(例如,使用深度缓冲器或类似布置)执行深度测试和剔除的电路。在一个实现方式中,光线追踪单元1527A-1527F与图像降噪协作地执行遍历和相交操作,该图像降噪的至少部分可使用相关联的矩阵加速单元1523A-1523F、1525A-1525D来执行。
图15C是根据本文中描述的实施例的通用图形处理单元(GPGPU)1570的框图,该GPGPU 1570可被配置为图形处理器(例如,图形处理器1508)和/或计算加速器。GPGPU 1570可经由一个或多个系统和/或存储器总线与主机处理器(例如,一个或多个CPU 1546)和存储器1571、1572互连。存储器1571可以是可与一个或多个CPU 1546进行共享的系统存储器,而存储器1572是专用于GPGPU 1570的设备存储器。例如,GPGPU 1570和存储器1572内的部件可被映射到能够由一个或多个CPU1546访问的存储器地址中。可经由存储器控制器1568来促进对存储器1571和1572的访问。存储器控制器1568可包括内部直接存储器存取(directmemory access,DMA)控制器1569,或可包括用于执行否则将由DMA控制器执行的操作的逻辑。
GPGPU 1570包括多个缓存存储器,这些缓存存储器包括L2缓存1553、L1缓存1554、指令缓存1555以及共享存储器1556,该共享存储器1556的至少部分也可被分区为缓存存储器。GPGPU 1570还包括多个计算单元1560A-1560N。每个计算单元1560A-1560N包括向量寄存器的集合1561、标量寄存器的集合1562、向量逻辑单元的集合1563以及标量逻辑单元的集合1564。计算单元1560A-1560N还可包括本地共享存储器1565和程序计数器1566。计算单元1560A-1560N可与常量缓存1567耦合,该常量缓存1567可用于存储常量数据,该常量数据是在GPGPU 1570上执行的内核程序或着色器程序的运行期间不会改变的数据。常量缓存1567可以是标量数据缓存,并且经缓存的数据可被直接取到标量寄存器1562中。
在操作期间,一个或多个CPU 1546可将命令写入到GPGPU1570中的寄存器中,或写入到GPGPU 1570中的、已经被映射到可访问地址空间中的存储器中。命令处理器1557可从寄存器或存储器读取命令,并且确定如何将在GPGPU 1570内处理那些命令。随后可使用线程调遣器1558来将线程调遣给计算单元1560A-1560N以执行那些命令。每个计算单元1560A-1560N可独立于其他计算单元来执行线程。此外,每个计算单元1560A-1560N可被独立地配置成用于有条件计算,并且可有条件地将计算的结果输出到存储器。当所提交的命令完成时,命令处理器1557可中断一个或多个CPU 1546。
图16A-图16C图示由本文中描述、例如根据图15A-图15C的实施例提供的附加的图形处理器和计算加速器体系结构的框图。图16A-图16C的具有与本文中任何其他附图的元件相同或类似名称的元件描述与其他附图中相同的元件,能以与其他附图中类似的方式进行操作或运行,可包括相同的部件,并且可链接到其他实体,该实体如本文中其他地方所描述的那些实体,但不限于此。
图16A是图形处理器1600的框图,该图形处理器1600可以是分立的图形处理单元,或可以是与多个处理核心或其他半导体器件集成的图形处理器,其他半导体器件诸如但不限于存储器设备或网络接口。图形处理器1600可以是图形处理器1508的变体,并且可替代图形处理器1508被使用。因此,本文中结合图形处理器1508对任何特征的公开也公开了对应的与图形处理器1600的结合,但不限于此。图形处理器可经由至图形处理器上的寄存器的存储器映射的I/O接口并且利用被放置到处理器存储器中的命令进行通信。图形处理器1600可包括用于访问存储器的存储器接口1614。存储器接口1614可以是至本地存储器、一个或多个内部缓存、一个或多个共享的外部缓存和/或至系统存储器的接口。
任选地,图形处理器1600还包括用于将显示输出数据驱动到显示设备1618的显示控制器1602。显示控制器1602包括用于显示器的一个或多个叠加平面以及多层的视频或用户界面元素的合成的硬件。显示设备1618可以是内部或外部显示设备。在一个实施例中,显示设备1618是头戴式显示设备,诸如,虚拟现实(VR)显示设备或增强现实(AR)显示设备。图形处理器1600可包括用于将媒体编码到一种或多种媒体编码格式,从一种或多种媒体编码格式对媒体解码,或在一种或多种媒体编码格式之间对媒体转码的视频编解码器引擎1606,这一种或多种媒体编码格式包括但不限于:移动图片专家组(Moving PictureExperts Group,MPEG)格式(诸如,MPEG-2)、高级视频译码(Advanced Video Coding,AVC)格式(诸如,H.264/MPEG-4AVC、H.265/HEVC、开放媒体联盟(Alliance for Open Media,AOMedia)VP8、VP9)、以及电影和电视工程师协会(the Society of Motion Picture&Television Engineers,SMPTE)421M/VC-1、和联合图像专家组(Joint PhotographicExperts Group,JPEG)格式(诸如,JPEG、以及运动JPEG(Motion JPEG,MJPEG)格式)。
图形处理器1600可包括块图像传输(block image transfer,BLIT)引擎1603,用于执行二维(2D)栅格化器操作,包括例如,比特边界块传输。然而,替代地,可使用图形处理引擎(graphics processing engine,GPE)1610的一个或多个部件执行2D图形操作。在一些实施例中,GPE1610是用于执行图形操作的计算引擎,这些图形操作包括三维(3D)图形操作和媒体操作。
GPE 1610可包括用于执行3D操作的3D管线1612,该3D操作诸如,使用作用于3D基元形状(例如,矩形、三角形等)的处理函数来渲染三维图像和场景。3D管线1612包括可编程和固定功能元件,这些可编程和固定功能元件执行元件内的各种任务和/或生成到3D/媒体子系统1615的执行线程。虽然3D管线1612可用于执行媒体操作,但是GPE 1610的实施例还包括媒体管线1616,该媒体管线1616专门用于执行媒体操作,诸如,视频后处理和图像增强。
媒体管线1616可包括固定功能或可编程逻辑单元,用于代替、或代表视频编解码器引擎1606来执行一个或多个专业的媒体操作,诸如,视频解码加速、视频去隔行以及视频编码加速。媒体管线1616可附加地包括线程生成单元,用于生成线程以供在3D/媒体子系统1615上执行。所生成的线程在3D/媒体子系统1615中所包括的一个或多个图形执行单元上执行用于媒体操作的计算。
3D/媒体子系统1615可包括用于执行由3D管线1612和媒体管线1616生成的线程的逻辑。管线可将线程执行请求发送到3D/媒体子系统1615,该3D/媒体子系统1615包括用于对于对可用的线程执行资源的各种请求进行仲裁和调遣的线程调遣逻辑。执行资源包括用于处理3D线程和媒体线程的图形执行单元的阵列。3D/媒体子系统1615可包括用于线程指令和数据的一个或多个内部缓存。此外,3D/媒体子系统1615还可包括用于在线程之间共享数据并用于存储输出数据的共享存储器,其包括寄存器和可寻址存储器。
图16B图示图形处理器1620,该图形处理器1620是图形处理器1600的变体,并且可替代图形处理器1600被使用且反之亦然。因此,本文中结合图形处理器1600对任何特征的公开也公开了对应的与图形处理器1620的结合,但不限于此。根据本文中描述的实施例,图形处理器1620具有分片体系结构。图形处理器1620可包括图形处理引擎集群1622,该图形处理引擎集群1622在图形引擎片1610A-1610D内具有图16A的图形处理器引擎1610的多个实例。每个图形引擎片1610A-1610D可经由片互连的集合1623A-1623F被互连。每个图形引擎片1610A-1610D还可经由存储器互连1625A-1625D被连接到存储器模块或存储器设备1626A-1626D。存储器设备1626A-1626D可使用任何图形存储器技术。例如,存储器设备1626A-1626D可以是图形双倍数据速率(GDDR)存储器。存储器设备1626A-1626D可以是高带宽存储器(HBM)模块,这些HBM模块可与其相应的图形引擎片1610A-1610D一起在管芯上。存储器设备1626A-1626D可以是可被堆叠在其相应的图形引擎片1610A-1610D的顶部上的堆叠式存储器设备。每个图形引擎片1610A-1610D和相关联的存储器1626A-1626D可驻留在分开的小芯片上,这些分开的小芯片被接合到基础管芯或基础衬底,如在图24B-图24D中进一步详细地所描述。
图形处理器1620可配置有非统一存储器存取(non-uniform memory access,NUMA)系统,在该NUMA系统中,存储器设备1626A-1626D与相关联的图形引擎片1610A-1610D耦合。给定的存储器设备可由与该存储器设备直接连接到的图形引擎片不同的图形引擎片访问。然而,当存取本地片时,对存储器设备1626A-1626D的存取等待时间可以最低。在一个实施例中,启用缓存一致的NUMA(cache coherent NUMA,ccNUMA)系统,该ccNUMA系统使用片互连1623A-1623F来启用图形引擎片1610A-1610D内的缓存控制器之间的通信,以便当多于一个缓存存储相同的存储器位置时保持一致的存储器图像。
图形处理引擎集群1622可与芯片上或封装上结构互连1624连接。在一个实施例中,结构互连1624包括网络处理器、片上网络(network on a chip,NoC)、或用于使结构互连1624能充当在图形处理器1620的部件之间交换数据分组的分组交换型结构互连的另一交换处理器。结构互连1624可启用图形引擎片1610A-1610D与诸如视频编解码器1606和一个或多个复制引擎1604之类的部件之间的通信。复制引擎1604可用于将数据移出存储器设备1626A-1626D和在图形处理器1620外部的存储器(例如,系统存储器),将数据移入存储器设备1626A-1626D和在图形处理器1620外部的存储器(例如,系统存储器),并且在存储器设备1626A-1626D与在图形处理器1620外部的存储器(例如,系统存储器)之间移动数据。结构互连1624还可用于将图形引擎片1610A-1610D互连。图形处理器1620可任选地包括显示控制器1602,用于启用与外部显示设备1618的连接。图形处理器还可被配置为图形加速器或计算加速器。在加速器配置中,显示控制器1602和显示设备1618可被省略。
图形处理器1620可经由主机接口1628连接到主机系统。主机接口1628可启用图形处理器1620、系统存储器和/或其他系统部件之间的通信。主机接口1628可以是例如PCI快速总线或另一类型的主机系统接口。例如,主机接口1628可以是NVLink或NVSwitch接口。主机接口1628和结构互连1624可以协作以使图形处理器1620的多个实例能充当单个逻辑设备。主机接口1628和结构互连1624之间的协作还可使各个图形引擎片1610A-1610D能够作为不同的逻辑图形设备向主机系统呈现。
图16C图示根据本文中描述的实施例的计算加速器1630。计算加速器1630可包括与图16B的图形处理器1620的体系结构类似性,并且针对计算加速进行优化。计算引擎集群1632可包括计算引擎片1640A-1640D的集合,计算引擎片1640A-1640D的集合包括针对并行或基于向量的通用计算操作优化的执行逻辑。计算引擎片1640A-1640D可以不包括固定功能图形处理逻辑,但是在一些实施例中,计算引擎片1640A-1640D中的一个或多个可包括用于执行媒体加速的逻辑。计算引擎片1640A-1640D可经由存储器互连1625A-1625D连接到存储器1626A-1626D。存储器1626A-1626D和存储器互连1625A-1625D可以是与在图形处理器1620中类似的技术,或者可以是不同的技术。计算引擎片1640A-1640D还可经由片互连的集合1623A-1623F被互连,并且可与结构互连1624连接和/或通过结构互连1624被互连。在一个实施例中,计算加速器1630包括可被配置为设备范围的缓存的大型L3缓存1636。计算加速器1630还能以与图16B的图形处理器1620类似的方式经由主机接口1628连接到主机处理器和存储器。
计算加速器1630还可包括集成的网络接口1642。在一个实施例中,集成的网络接口1642包括网络处理器和控制器逻辑,该控制器逻辑使计算引擎集群1632能够在无需数据跨越主机系统的存储器的情况下通过物理层互连1644进行通信。在一个实施例中,计算引擎片1640A-1640D中的一个由网络处理器逻辑替代,并且要经由物理层互连1644传送或接收的数据可直接向存储器1626A-1626D或从存储器1626A-1626D传送。计算加速器1630的多个实例可经由物理层互连1644被结合到单个逻辑设备中。替代地,各计算引擎片1640A-1640D可被呈现为不同的网络可访问计算加速器设备。
图形处理引擎
图17是根据一些实施例的图形处理器的图形处理引擎1710的框图。图形处理引擎(GPE)1710可以是图16A中示出的GPE 1610的某个版本,并且还可表示图16B的图形引擎片1610A-1610D。图17的具有与本文中任何其他附图的元件相同或类似名称的元件描述与其他附图中相同的元件,能以与其他附图中类似的方式进行操作或运行,可包括相同的部件,并且可链接到其他实体,该实体如本文中其他地方所描述的那些实体,但不限于此。例如,在图17中也图示图16A的3D管线1612和媒体管线1616。媒体管线1616在GPE 1710的一些实施例中是任选的,并且可以不显式地被包括在GPE 1710内。例如并且在至少一个实施例中,单独的媒体和/或图像处理器被耦合至GPE 1710。
GPE 1710可与命令流转化器1703耦合或包括命令流转化器1703,该命令流转化器1703将命令流提供给3D管线1612和/或媒体管线1616。替代地或附加地,命令流转化器1703可直接耦合至统一返回缓冲器1718。统一返回缓冲器1718可通信地耦合至图形核心集群1714。任选地,命令流转化器1703与存储器耦合,该存储器可以是系统存储器、或内部缓存存储器和共享缓存存储器中的一个或多个。命令流转化器1703可从存储器接收命令,并且将这些命令发送至3D管线1612和/或媒体管线1616。这些命令是从环形缓冲器取得的指示,该环形缓冲器存储用于3D管线1612和媒体管线1616的命令。环形缓冲器可附加地包括存储批量的多个命令的批量命令缓冲器。用于3D管线1612的命令还可包括对存储在存储器中的数据的引用,这些数据诸如但不限于用于3D管线1612的顶点数据和几何数据和/或用于媒体管线1616的图像数据和存储器对象。3D管线1612和媒体管线1616通过经由相应的管线内的逻辑执行操作或者通过将一个或多个执行线程调遣至图形核心集群1714来处理命令和数据。图形核心集群1714可包括一个或多个图形核心块(例如,图形核心块1715A、图形核心块1715B),每个块包括一个或多个图形核心。每个图形核心包括图形执行资源的集合,该图形执行资源的集合包括:用于执行图形操作和计算操作的通用和图形专用执行逻辑;以及固定功能纹理处理逻辑和/或机器学习和人工智能加速逻辑。
在各实施例中,3D管线1612可包括用于通过处理指令并将执行线程调遣给图形核心集群1714来处理一个或多个着色器程序的固定功能和可编程逻辑,该一个或多个着色器程序诸如,顶点着色器、几何着色器、像素着色器、片段着色器、计算着色器或其他着色器程序。图形核心集群1714提供统一的执行资源块,以供在处理这些着色器程序时使用。图形核心集群1714的图形核心块1715A-1715B内的多功能执行逻辑(例如,执行单元)包括对各种3D API着色器语言的支持,并且可执行与多个着色器相关联的多个同步执行线程。
图形核心集群1714可包括用于执行诸如视频和/或图像处理之类的媒体功能的执行逻辑。除了图形处理操作之外,执行单元还可包括可编程以执行并行的通用计算操作的通用逻辑。通用逻辑可并行地或结合图14的(一个或多个)处理器核心1407或如图15A中的核心1502A-1502N内的通用逻辑来执行处理操作。
由在图形核心集群1714上执行的线程生成的输出数据可以将数据输出到统一返回缓冲器(unified return buffer,URB)1718中的存储器。URB 1718可存储用于多个线程的数据。URB 1718可用于在图形核心集群1714上执行的不同线程之间发送数据。URB 1718可附加地用于在图形核心集群1714上的线程与共享功能逻辑1720内的固定功能逻辑之间的同步。
任选地,图形核心集群1714可以是可缩放的,使得阵列包括可变数量的图形核心,每个图形核心都具有基于GPE 1710的目标功率和性能等级的可变数量的执行单元。执行资源可以是动态地可缩放的,使得执行资源可根据需要被启用或禁用。
图形核心集群1714与共享功能逻辑1720耦合,该共享功能逻辑1720包括在图形核心阵列中的图形核心之间被共享的多个资源。共享功能逻辑1720内的共享功能是将专业的补充功能提供给图形核心集群1714的硬件逻辑单元。在各实施例中,共享功能逻辑1720包括但不限于采样器1721逻辑、数学1722逻辑和线程间通信(inter-thread communication,ITC)1723逻辑。此外,可实现共享功能逻辑1720内的一个或多个缓存1725。
至少在其中对于给定的专业功能的需求不足以包括在图形核心集群1714内的情况下实现共享功能。相反,那个专业功能的单个实例化被实现为共享功能逻辑1720中的独立实体,并且在图形核心集群1714内的执行资源之间被共享。在图形核心集群1714之间被共享并被包括在图形核心集群1714内的确切的功能集因实施例而异。共享功能逻辑1720内的由图形核心集群1714广泛使用的特定共享功能可被包括在图形核心集群1714内的共享功能逻辑1716内。任选地,图形核心集群1714内的共享功能逻辑1716可包括共享功能逻辑1720内的一些或所有逻辑。共享功能逻辑1720内的所有逻辑元件可以在图形核心集群1714的共享功能逻辑1716内被复制。替代地,共享功能逻辑1720被排除以有利于图形核心集群1714内的共享功能逻辑1716。
图形处理资源
图18A-图18C图示根据本文中描述的实施例的包括在图形处理器中采用的处理元件阵列的执行逻辑。图18A图示根据实施例的图形核心集群。图18B图示根据实施例的图形核心的向量引擎。图18C图示根据实施例的图形核心的矩阵引擎。图18A-图18C的具有与本文中任何其他附图的元件相同的附图标记的元件能以与在本文中其他地方描述的方式类似的任何方式进行操作或运行,但不限于此。例如,图18A-图18C的元件能以图15B的图形处理器核心块1519和/或图17的图形核心块1715A-1715B的上下文来考虑。在一个实施例中,图18A-图18C的元件具有与图15A的图形处理器1508或图15C的GPGPU 1570的等效部件类似的功能。
如图18A中所示,在一个实施例中,图形核心集群1714包括图形核心块1715,该图形核心块1715可以是图17的图形核心块1715A或图形核心块1715B。图形核心块1715可包括任何数量的图形核心(例如,图形核心1815A、图形核心1815B,一直到图形核心1815N)可以包括图形核心块1715的多个实例。在一个实施例中,图形核心1815A-1815N的元件具有与图15B的图形核心1521A-1521F的元件类似或等效的功能。在此类实施例中,图形核心1815A-1815N各自包括电路,包括但不限于:向量引擎1802A-1802N、矩阵引擎1803A-1803N、存储器加载/存储单元1804A-1804N、指令缓存1805A-1805N、数据缓存/共享本地存储器1806A-1806N、光线追踪单元1808A-1808N、采样器1810A-1810N。图形核心1815A-1815N的电路可以附加地包括固定功能逻辑1812A-1812N。设计的图形核心1815A-1815N内的向量引擎1802A-1802N和矩阵引擎1803A-1803N的数量可以基于针对该设计的工作负载、性能和功率目标而变化。
参考图形核心1815A,向量引擎1802A和矩阵引擎1803A可配置用于基于与着色器程序相关联的指令对以各种整数和浮点数据格式的数据执行并行的计算操作。每个向量引擎1802A和矩阵引擎1803A可以充当能够执行多个同步硬件线程同时针对每个线程并行地处理多个数据元素的可编程通用计算单元。向量引擎1802A和矩阵引擎1803A支持处理处于各种SIMD宽度的可变宽度向量,包括但不限于SIMD8、SIMD16和SIMD32。输入数据元素可以作为紧缩数据类型存储在寄存器中,并且向量引擎1802A和矩阵引擎1803A可基于元素的数据大小来处理各元素。例如,当对256比特宽的向量进行操作时,向量的256比特被存储在寄存器中,并且向量被处理为四个单独的64比特紧缩数据元素(四字(Quad-Word,QW)大小数据元素)、八个单独的32比特紧缩数据元素(双字(Double Word,DW)大小数据元素)、十六个单独的16比特紧缩数据元素(字(Word,W)大小数据元素)、或三十二个单独的8比特数据元素(字节(byte,B)大小数据元素)。然而,不同的向量宽度和寄存器大小是可能的。在一个实施例中,向量引擎1802A和矩阵引擎1803A还可配置用于对各大小的单元组和线程组(例如,8个、16个或32个线程)进行SIMT操作。
继续图形核心1815A,存储器加载/存储单元1804A服务于存储器访问请求,该存储器访问请求由向量引擎1802A、矩阵引擎1803A和/或图形核心1815A的具有对存储器的访问权的其他部件来发出。存储器访问请求可以由存储器加载/存储单元1804A处理,以将所请求的数据加载或存储到缓存或存储器、或从缓存或存储器加载或存储到与向量引擎1802A和/或矩阵引擎1803A相关联的寄存器堆中。存储器加载/存储单元1804A还可执行预取操作。另外参考图19,在一个实施例中,存储器加载/存储单元1804A被配置用于针对被存储在存储器1910中、来自经由片互连1908而对于其他片而言是本地的存储器、或者来自系统存储器的数据提供SIMT分散/聚集预取或块预取。可以对特定L1缓存(例如,数据缓存/共享本地存储器1806A)、L2缓存1904或L3缓存1906执行预取。在一个实施例中,对L3缓存1906的预取自动地导致数据被存储在L2缓存1904中。
指令缓存1805A存储要由图形核心1815A执行的指令。在一个实施例中,图形核心1815A还包括将指令取到或预取到指令缓存1805A中的指令取得和预取电路。图形核心1815A还包括指令解码逻辑,以用于对指令缓存1805A内的指令进行解码。数据缓存/共享本地存储器1806A可以被配置为由实现缓存替代策略的缓存控制器管理的数据缓存并且/或者被配置为被显式地管理的共享存储器。光线追踪单元1808A包括用于加速光线追踪操作的电路。采样器1810A为3D操作提供纹理采样,并为媒体操作提供媒体采样。固定功能逻辑1812A包括固定功能电路,该固定功能电路在向量引擎1802A和矩阵引擎1803A的各实例之间共享。图形核心1815B-1815N能以与图形核心1815A类似的方式进行操作。
指令缓存1805A-1805N、数据缓存/共享本地存储器1806A-1806N、光线追踪单元1808A-1808N、采样器1810A-1812N和固定功能逻辑1812A-1812N的功能与本文中描述的图形处理器体系结构中的等效功能相对应。例如,指令缓存1805A-1805N能以与图15C的指令缓存1555类似的方式进行操作。数据缓存/共享本地存储器1806A-1806N、光线追踪单元1808A-1808N和采样器1810A-1812N能以与图15B的缓存/SLM 1528A-1528F、光线追踪单元1527A-1527F和采样器1526A-1526F类似的方式进行操作。固定功能逻辑1812A-1812N可以包括图15B的几何/固定功能管线1531和/或附加的固定功能逻辑1538的元件。在一个实施例中,光线追踪单元1808A-1808N包括用于执行由图3C的光线追踪核心372所执行的光线追踪加速操作的电路。
如图18B中所示,在一个实施例中,向量引擎1802包括指令取得单元1837、通用寄存器堆阵列(general register file,GRF)1824、体系结构寄存器堆阵列(architecturalregister file,ARF)1826、线程仲裁器1822、发送单元1830、分支单元1832、SIMD浮点单元(FPU)的集合1834、以及在一个实施例中的整数SIMD ALU的集合1835。GRF 1824和ARF1826包括与可在向量引擎1802中活跃的每个硬件线程相关联的通用寄存器堆和体系结构寄存器堆的集合。在一个实施例中,每线程体系结构状态被维持在ARF 1826中,而在线程执行期间使用的数据被存储在GRF 1824中。每个线程的执行状态,包括用于每个线程的指令指针,可以被保存在ARF 1826中的线程特定寄存器中。可使用寄存器重命名来动态地将寄存器分配给硬件线程。
在一个实施例中,向量引擎1802具有作为同步多线程(Simultaneous Multi-Threading,SMT)与细粒度交织多线程(Interleaved Multi-Threading,IMT)的组合的体系结构。该体系结构具有模块化配置,该模块化配置可以基于同步线程的目标数量和每个图形核心的寄存器的数量而在设计时进行微调,其中跨用于执行多个同步线程的逻辑来划分图形核心资源。可由向量引擎1802执行的逻辑线程的数量不限于硬件线程的数量,并且可将多个逻辑线程指派给每个硬件线程。
在一个实施例中,向量引擎1802可协同发出多个指令,这些指令可以各自是不同的指令。线程仲裁器1822可以将指令调遣到发送单元1830、分支单元1832或(一个或多个)SIMD FPU 1834中的一个以供执行。每个执行线程可访问GRF 1824内的128个通用寄存器,其中,每个寄存器可存储可作为具有32字节数据元素的可变宽度向量访问的32个字节。在一个实施例中,每个线程具有对GRF 1824内的4个千字节的访问权,但是实施例并不限于此,并且在其他实施例中可以提供更多或更少的寄存器资源。在一个实施例中,向量引擎1802被分区为可独立地执行计算操作的七个硬件线程,但是每个向量引擎1802的线程数量也可根据实施例而有所不同。例如,在一个实施例中,支持最多16个硬件线程。在其中七个线程可以访问4个千字节的实施例中,GRF 1824可以存储总共28个千字节。在16个线程可访问4个千字节的情况下,GRF 1824可存储总共64个千字节。灵活的寻址模式可准许对寄存器一起进行寻址,从而有效地建立更宽的寄存器或者表示跨步式矩形块数据结构。
在一个实施例中,经由由消息传递发送单元1830执行的“发送”指令来调遣存储器操作、采样器操作以及其他较长等待时间的系统通信。在一个实施例中,分支指令被调遣给专用分支单元1832,以促进SIMD分散和最终的汇聚。
在一个实施例中,向量引擎1802包括用于执行浮点操作的一个或多个SIMD浮点单元((一个或多个)FPU)1834。在一个实施例中,(一个或多个)FPU 1834还支持整数计算。在一个实施例中,(一个或多个)FPU 1834可以执行最多M个32比特浮点(或整数)操作,或者执行最多2M个16比特整数或16比特浮点操作。在一个实施例中,(一个或多个)FPU中的至少一个提供支持高吞吐量超越数学函数和双精度64比特浮点的扩展数学能力。在一些实施例中,8比特整数SIMD ALU的集合1835也存在,并且可专门优化成执行与机器学习计算相关联的操作。在一个实施例中,SIMD ALU由可配置用于执行整数和浮点操作的附加的SIMD ALU的集合1834来替代。在一个实施例中,SIMD FPU 1834和SIMD ALU 1835可配置用于执行SIMT程序。在一个实施例中,支持组合的SIMD+SIMT操作。
在一个实施例中,向量引擎1802的多个实例的阵列可在图形核心中被实例化。为了可缩放性,产品架构师可以选择每图形核心分组的向量引擎的确切数量。在一个实施例中,向量引擎1802可以跨多个执行通道来执行指令。在进一步的实施例中,在不同通道上执行在向量引擎1802上执行的每个线程。
如图18C中所示,在一个实施例中,矩阵引擎1803包括被配置用于执行张量操作的处理元件的阵列,该张量操作包括向量/矩阵操作和矩阵/矩阵操作,诸如但不限于矩阵乘法和/或点积操作。可以利用M行和N列的处理元件(1852AA-1852MN)来配置矩阵引擎1803,该处理元件(PE 1852AA-PE 1852MN)包括以管线化方式组织的乘法器和加法器电路。在一个实施例中,处理元件1852AA-1852MN组成N宽和M深的脉动阵列的物理管线阶段,该脉动阵列可用于以数据并行的方式执行向量/矩阵操作或矩阵/矩阵操作,包括矩阵乘法、融合乘加、点积或其他通用矩阵-矩阵乘法(GEMM)操作。在一个实施例中,矩阵引擎1803支持16比特和8比特的浮点操作,以及8比特、4比特、2比特和二进制整数操作。矩阵引擎1803还可以被配置用于加速特定机器学习操作。在此类实施例中,矩阵引擎1803可配置有对于相对于电气和电子工程师学会(Institute of Electrical and Electronics Engineers,IEEE)754格式具有不同数量的尾数比特和指数比特的bfloat(brain浮点)16比特浮点格式、或张量浮点32比特浮点格式(TF32)的支持。
在一个实施例中,在每个周期期间,每个阶段可以将在该阶段执行的操作的结果添加至前一阶段的输出。在其他实施例中,在计算周期的集合之后,处理元件1852AA-1852MN之间的数据移动的模式可以基于被执行的指令或宏操作而变化。例如,在一个实施例中,部分和回路(partial sum loopback)被启用,并且处理元件可以替代地将当前周期的输出与前一周期中生成的输出相加。在一个实施例中,脉动阵列的最终阶段可被配置有到脉动阵列的初始阶段的回路。在此类实施例中,物理管线阶段的数量可以与由矩阵引擎1803所支持的逻辑管线阶段的数量解耦。例如,在处理元件1852AA-1852MN被配置为M个物理阶段的脉动阵列的情况下,从阶段M至初始管线阶段的回路可以使得处理元件1852AA-1852MN能够作为例如2M、3M、4M等的逻辑管线阶段的脉动阵列来进行操作。
在一个实施例中,矩阵引擎1803包括存储器1841A-1841N、1842A-1842M,用于以针对输入矩阵的行和列数据的形式存储输入数据。存储器1842A-1842M可配置用于存储第一输入矩阵的行元素(A0-Am),并且存储器1841A-1841N可配置用于存储第二输入矩阵的列元素(B0-Bn)。行元素和列元素被提供为到处理元件1852AA-1852MN的输入以供处理。在一个实施例中,输入矩阵的元素行和列元素可以在这些元素被提供给存储器1841A-1841N、1842A-1842M之前存储在矩阵引擎1803内的脉动寄存器堆1840中。在一个实施例中,排除脉动寄存器堆1840,并且从相关联的向量引擎中的寄存器(例如,图18B的向量引擎1802的GRF1824)或包括矩阵引擎1803的图形核心的其他存储器(例如,图18A的用于矩阵引擎1803A的数据缓存/共享本地存储器1806A)加载存储器1841A-1841N、1842A-1842M。由处理元件1852AA-1852MN生成的结果随后被输出到输出缓冲器和/或被写入到寄存器堆(例如,脉动寄存器堆1840、GRF 1824、数据缓存/共享本地存储器1806A-1806N),以供图形处理器的其他功能单元进一步处理或供输出到存储器。
在一些实施例中,矩阵引擎1803被配置有对输入稀疏度的支持,其中,输入数据的稀疏区域的乘法操作可通过跳过具有零值的操作对象的乘法操作而被绕过。在一个实施例中,处理元件1852AA-1852MN被配置用于跳过具有零值输入的某些操作的执行。在一个实施例中,输入矩阵内的稀疏度可以被检测,并且具有已知零输出值的操作在被提交给处理元件1852AA-1852MN之前可以被绕过。将零值操作对象加载到处理元件中可以被绕过,并且处理元件1852AA-1852MN可被配置用于对非零值输入元素执行乘法。矩阵引擎1803还可被配置有对输出稀疏度的支持,使得具有被预定为零的结果的操作可被绕过。对于输入稀疏度和/或输出稀疏度,在一个实施例中,将元数据提供给处理元件1852AA-1852MN,以指示对于某一处理周期,哪些处理元件和/或数据通道在该周期期间将是活跃的。
在一个实施例中,矩阵引擎1803包括用于启用对具有稀疏矩阵的压缩表示的稀疏数据的操作的硬件,该稀疏矩阵存储非零值和标识该非零值在矩阵内的位置的元数据。示例性压缩表示包括但不限于压缩张量表示,诸如,压缩稀疏行(CSR)表示、压缩稀疏列(CSC)表示、压缩稀疏纤维(compressed sparse fiber,CSF)表示。对压缩表示的支持使得操作能够对按压缩张量格式的输入执行而无需压缩表示被解压缩或解码。在此类实施例中,可仅对非零输入值执行操作,并且所得到的非零输出值可被映射到输出矩阵中。在一些实施例中,还提供对机器特定无损数据压缩格式的硬件支持,这些机器特定无损数据压缩格式当在硬件内传送数据或跨系统总线传送数据时被使用。此类数据可按用于稀疏输入数据的压缩格式被保留,并且矩阵引擎1803可使用用于经压缩数据的压缩元数据,以使得操作能够仅对非零值执行或使得对于乘法操作能够绕过零数据输入的块。
在各种实施例中,可以由编程器以压缩张量表示来提供输入数据,或者编解码器可以将输入数据压缩为压缩张量表示或另一稀疏数据编码。此外,为了支持压缩张量表示,可在输入数据被提供到处理元件1852AA-1852MN之前执行稀疏输入数据的流式压缩。在一个实施例中,对被写入到与图形核心集群1714相关联的缓存存储器的数据执行压缩,其中利用由矩阵引擎1803所支持的编码来执行该压缩。在一个实施例中,矩阵引擎1803包括对具有结构化稀疏度的输入的支持,在该结构化稀疏度中,预定级别或预定模式的稀疏度被施加在输入数据上。该数据可被压缩到已知的压缩率,其中,经压缩的数据由压缩元件1852AA-1852MN根据与经压缩的数据相关联的元数据来处理。
图19图示根据实施例的多片处理器的片1900。在一个实施例中,片1900表示图16B的图形引擎片1610A-1610D或图16C的计算引擎片1640A-1640D中的一个。多片图形处理器的片1900包括图形核心集群的阵列(例如,图形核心集群1714A、图形核心集群1714B,一直到图形核心集群1714N),其中,每个图形核心集群具有图形核心的阵列1815A-1815N。片1900还包括全局调遣器1902,用于将线程调遣到片1900的处理资源。
片1900可以包括L3缓存1906和存储器1910或与L3缓存1906和存储器1910耦合。在各实施例中,可以排除L3缓存1906,或者片1900可以包括附加级别的缓存,诸如L4缓存。在一个实施例中,诸如图16B和图16C中,多片图形处理器中的片1900的每个实例具有相关联的存储器1910。在一个实施例中,多片处理器可以被配置为多芯片模块,在该多芯片模块中,L3缓存1906和/或存储器1910驻留在与图形核心集群1714A-1714N不同的单独的小芯片上。在该上下文中,小芯片是至少部分地被封装的集成电路,该至少部分地被封装的集成电路包括能够与其他小芯片一起被组装到更大的封装中的不同的逻辑单元。例如,L3缓存1906可被包括在专用缓存小芯片中,或驻留在与图形核心集群1714A-1714N相同的小芯片上。在一个实施例中,L3缓存1906可被包括在如图24C所图示的活跃的基础管芯或活跃的中介层中。
存储器结构1903启用图形核心集群1714A-1714N、L3缓存1906和存储器1910之间的通信。L2缓存1904与存储器结构1903耦合,并且可配置用于缓存经由存储器结构1903执行的事务。片互连1908启用与图形处理器上的其他片的通信,并且可以是图16B和图16C的片互连1623A-1623F中的一个。在从片1900排除L3缓存1906的实施例中,L2缓存1904可以被配置为组合的L2/L3缓存。存储器结构1903可配置用于基于L3缓存1906存在于或不存在于特定实现方式中,而将数据路由到L3缓存1906或路由到与存储器1910相关联的存储器控制器。L3缓存1906可被配置为逐片(per-tile)缓存,该逐片缓存专用于片1900的处理资源或者可以是GPU宽的L3缓存的部分。
图20是图示图形处理器指令格式2000的框图。图形处理器执行单元支持具有按照多种格式的指令的指令集。实线框图示通常被包括在执行单元指令中的组成部分,而虚线包括任选的或仅被包括在指令的子集中的组成部分。在一些实施例中,所描述和图示的图形处理器指令格式2000是宏指令,因为它们是供应至执行单元的指令,这与产生自一旦指令被处理就进行的指令解码的微操作相反。因此,单个指令可使硬件执行多个微操作。
如本文中所描述的图形处理器执行单元可以原生地支持128比特指令格式2010的指令。基于所选择的指令、指令选项和操作对象数量,64比特紧凑指令格式2030可用于一些指令。原生的128比特指令格式2010提供对所有指令选项的访问,而一些选项和操作在64比特格式2030中受限。64比特格式2030中可用的原生指令因实施例而异。使用索引字段2013中的索引值的集合将指令部分地压缩。执行单元硬件基于索引值来引用压缩表的集合,并使用压缩表输出来重构128比特指令格式2010的原生指令。可以使用其他大小和格式的指令。
针对每种格式,指令操作码2012限定执行单元要执行的操作。执行单元跨每个操作对象的多个数据元素并行地执行每个指令。例如,响应于加法指令,执行单元跨表示纹理元素或图片元素的每个颜色通道执行同步加法操作。默认地,执行单元跨操作对象的所有数据通道执行每个指令。指令控制字段2014可启用对某些执行选项(诸如,通道选择(例如,谓词(predication))和数据通道顺序(例如,拌和(swizzle)))的控制。针对128比特指令格式2010的指令,执行大小字段2016限制将被并行地执行的数据通道的数量。执行大小字段2016可能不可用于64比特紧凑指令格式2030。
一些执行单元指令具有最多三个操作对象,包括两个源操作对象src0 2020、src12022以及一个目的地操作对象(dest 2018)。其他指令(诸如,例如数据操纵指令、点积指令、乘加指令、或乘法累加指令)可具有第三源操作对象(例如,SRC2 2024)。指令操作码2012确定源操作对象的数量。指令的最后一个源操作对象可以是与指令一起被传递的立即数(例如,硬编码的)值。执行单元还可以支持多个目的地指令,其中目的地中的一个或多个是基于指令和/或所指定的目的地而隐含的或隐式的。
128比特指令格式2010可包括访问/寻址模式字段2026,该访问/寻址模式字段2026例如指定使用直接寄存器寻址模式还是间接寄存器寻址模式。当使用直接寄存器寻址模式时,由指令中的比特直接提供一个或多个操作对象的寄存器地址。
128比特指令格式2010还可包括访问/寻址模式字段2026,该访问/寻址模式字段2026指定指令的寻址模式和/或访问模式。访问模式可用于限定指令的数据访问对齐。可支持包括16字节对齐访问模式和1字节对齐访问模式的访问模式,其中,访问模式的字节对齐确定指令操作对象的访问对齐。例如,当处于第一模式时,指令可将字节对齐的寻址用于源操作对象和目的地操作对象,并且当处于第二模式时,指令可将16字节对齐的寻址用于所有的源操作对象和目的地操作对象。
访问/寻址模式字段2026的寻址模式部分可以确定指令要使用直接寻址还是间接寻址。当使用直接寄存器寻址模式时,指令中的比特直接提供一个或多个操作对象的寄存器地址。当使用间接寄存器寻址模式时,可以基于指令中的地址寄存器值和地址立即数字段来计算一个或多个操作对象的寄存器地址。
可以基于操作码2012比特字段对指令进行分组从而简化操作码解码2040。针对8比特的操作码,比特4、比特5、和比特6允许执行单元确定操作码的类型。所示出的确切的操作码分组仅是示例。移动和逻辑操作码组2042可以包括数据移动和逻辑指令(例如,移动(mov)、比较(cmp))。移动和逻辑组2042可以共享五个最低有效的比特(leastsignificantbit,LSB),其中,移动(mov)指令采用0000xxxxb的形式,而逻辑指令采用0001xxxxb的形式。流控制指令组2044(例如,调用(call)、跳转(jmp))包括0010xxxxb(例如,0x20)形式的指令。混杂指令组2046包括指令的混合,包括0011xxxxb(例如,0x30)形式的同步指令(例如,等待(wait)、发送(send))。并行数学指令组2048包括0100xxxxb(例如,0x40)形式的逐分量的算术指令(例如,加、乘(mul))。并行数学指令组2048跨数据通道并行地执行算术操作。向量数学组2050包括0101xxxxb(例如,0x50)形式的算术指令(例如,dp4)。向量数学组对向量操作对象执行算术,诸如,点积计算。在一个实施例中,所图示的操作码解码2040可用于确定执行单元的哪个部分将用于执行经解码的指令。例如,一些指令可被指定为将由脉动阵列执行的脉动指令。其他指令(诸如,光线追踪指令(未示出))可被路由至执行逻辑的切片或分区内的光线追踪核心或光线追踪逻辑。
图形管线
图21是根据另一个实施例的图形处理器2100的框图。图21的具有与本文中任何其他附图的元件相同或类似名称的元件描述与其他附图中相同的元件,能以与其他附图中类似的方式进行操作或运行,可包括相同的部件,并且可链接到其他实体,该实体如本文中其他地方所描述的那些实体,但不限于此。
图形处理器2100可包括不同类型的图形处理管线,诸如,几何管线2120、媒体管线2130、显示引擎2140、线程执行逻辑2150、以及渲染输出管线2170。图形处理器2100可以是包括一个或多个通用处理核心的多核心处理系统内的图形处理器。图形处理器可通过至一个或多个控制寄存器(未示出)的寄存器写入或者经由通过环形互连2102发出至图形处理器2100的命令被控制。环形互连2102可将图形处理器2100耦合至其他处理部件(诸如,其他图形处理器或通用处理器)。由命令流转化器2103解释来自环形互连2102的命令,该命令流转化器2103将指令供应至几何管线2120或媒体管线2130的各个部件。
命令流转化器2103可引导顶点取得器2105的操作,该顶点取得器2105从存储器读取顶点数据,并执行由命令流转化器2103提供的顶点处理命令。顶点取得器2105可将顶点数据提供给顶点着色器2107,该顶点着色器2107对每一个顶点执行坐标空间变换和照明操作。顶点取得器2105和顶点着色器2107可通过经由线程调遣器2131将执行线程调遣给图形核心2152A-2152B来执行顶点处理指令。
图形核心2152A-2152B可以是具有用于执行图形操作和媒体操作的指令集的向量处理器的阵列。图形核心2152A-2152B可具有专用于每个阵列或在阵列之间被共享的所附接的L1缓存2151。缓存可以被配置为数据缓存、指令缓存、或被分区为在不同分区中包含数据和指令的单个缓存。
几何管线2120可包括用于执行3D对象的硬件加速曲面细分的曲面细分部件。可编程壳体着色器2111可配置曲面细分操作。可编程域着色器2117可提供对曲面细分输出的后端评估。曲面细分器2113可在外壳着色器2111的指示下进行操作,并且可包含用于基于粗糙的几何模型来生成详细的几何对象集合的专用逻辑,该粗糙的几何模型作为输入被提供给几何管线2120。此外,如果不使用曲面细分,则可以绕过曲面细分部件(例如,外壳着色器2111、曲面细分器2113和域着色器2117)。曲面细分部件可基于从顶点着色器2107接收的数据进行操作。
完整的几何对象可由几何着色器2119经由被调遣给图形核心2152A-2152B的一个或多个线程来处理,或者可以直接行进至裁剪器2129。几何着色器可对整个几何对象操作,而不是像在图形管线的先前的阶段中那样对顶点或顶点的补片进行操作。如果曲面细分被禁用,则几何着色器2119从顶点着色器2107接收输入。几何着色器2119可以是可由几何着色器程序编程的,以便在曲面细分单元被禁用的情况下执行几何曲面细分。
在栅格化之前,裁剪器2129处理顶点数据。裁剪器2129可以是固定功能裁剪器或具有裁剪和几何着色器功能的可编程裁剪器。渲染输出管线2170中的栅格化器和深度测试部件2173可调遣像素着色器以将几何对象转换为逐像素表示。像素着色器逻辑可被包括在线程执行逻辑2150中。任选地,应用可绕过栅格化器和深度测试部件2173,并且经由流出单元2123访问未栅格化的顶点数据。
图形处理器2100具有互连总线、互连结构、或允许数据和消息在处理器的主要部件之间传递的某个其他互连机制。在一些实施例中,图形核心2152A-2152B和相关联的逻辑单元(例如,L1缓存2151、采样器2154、纹理缓存2158等)经由数据端口2156进行互连,以执行存储器访问并且与处理器的渲染输出管线部件进行通信。采样器2154、缓存2151、2158和图形核心2152A-2152B各自可具有单独的存储器访问路径。任选地,纹理缓存2158也可被配置为采样器缓存。
渲染输出管线2170可包含栅格化器和深度测试部件2173,该栅格化器和深度测试部件2173将基于顶点的对象转换为相关联的基于像素的表示。栅格化器逻辑可包括用于执行固定功能三角形和线栅格化的窗口器/掩码器单元。在一些实施例中,相关联的渲染缓存2178和深度缓存2179也是可用的。像素操作部件2177对数据执行基于像素的操作,但是在一些实例中,与2D操作相关联的像素操作(例如,利用混合的比特块图像传输)由2D引擎2141执行,或者在显示时由显示控制器2143使用叠加显示平面来代替。共享的L3缓存2175可以可用于所有的图形部件,从而允许在不使用主系统存储器的情况下共享数据。
媒体管线2130可包括媒体引擎2137和视频前端2134。视频前端2134可从命令流转化器2103接收管线命令。媒体管线2130可包括单独的命令流转化器。视频前端2134可在将媒体命令发送到媒体引擎2137之前处理该媒体命令。媒体引擎2137可包括用于生成线程以用于经由线程调遣器2131调遣给线程执行逻辑2150的线程生成功能。
图形处理器2100可包括显示引擎2140。该显示引擎2140可在处理器2100外部,并且可经由环形互连2102、或某个其他互连总线或结构来与图形处理器耦合。显示引擎2140可包括2D引擎2141和显示控制器2143。显示引擎2140可包含能够独立于3D管线进行操作的专用逻辑。显示控制器2143可与显示设备(未示出)耦合,该显示设备可以是如在膝上型电脑中的系统集成的显示设备或经由显示设备连接器而附连的外部显示设备。
几何管线2120和媒体管线2130可以可被配置成用于基于多个图形和媒体编程接口来执行操作,并且不专用于任何一个应用编程接口(API)。用于图形处理器的驱动器软件可将专用于特定图形或媒体库的API调用转换为可由图形处理器处理的命令。可以为全部来自Khronos Group的开放图形库(Open Graphics Library,OpenGL)、开放计算语言(OpenComputing Language,OpenCL)和/或Vulkan图形和计算API提供支持。也可以为来自微软公司的Direct3D库提供支持。可支持这些库的组合。还可以为开源计算机视觉库(OpenSource Computer Vision Library,OpenCV)提供支持。如果可进行从未来API的管线到图形处理器的管线的映射,则具有兼容3D管线的未来API也将受到支持。
图形管线编程
图22A是图示用于对图形处理管线编程的图形处理器命令格式2200的框图,图形处理管线诸如例如本文中结合图16A、图17、图21描述的管线。图22B是图示根据实施例的图形处理器命令序列2210的框图。图22A中的实线框图示一般被包括在图形命令中的组成部分,而虚线包括任选的或仅被包括在图形命令的子集中的组成部分。图22A的示例性图形处理器命令格式2200包括用于标识命令的客户端2202、命令操作代码(操作码)2204和数据字段2206的字段。子操作码2205和命令大小2208也被包括在一些命令中。
客户端2202可指定图形设备的、处理命令数据的客户端单元。图形处理器命令解析器可检查每一个命令的客户端字段以调整对命令的进一步的处理,并且将命令数据路由至适当的客户端单元。图形处理器客户端单元可包括存储器接口单元、渲染单元、2D单元、3D单元和媒体单元。每个客户端单元可具有处理命令的对应的处理管线。一旦由客户端单元接收到命令,客户端单元就读取操作码2204以及子操作码2205(如果存在)以确定要执行的操作。客户端单元使用数据字段2206中的信息来执行命令。针对一些命令,预期显式的命令大小2208指定命令的大小。命令解析器可基于命令操作码自动地确定命令中的至少一些命令的大小。命令可经由双字的倍数被对齐。还可使用其他命令格式。
图22B中的流程图示示例性图形处理器命令序列2210。以示例性图形处理器为特征的数据处理系统的软件或固件可使用所示出的命令序列的某个版本来建立、执行并终止图形操作的集合。仅出于示例目的示出并描述样本命令序列,并且样本命令序列不限于这些特定的命令或该命令序列。此外,命令可以作为批量的命令在命令序列中被发出,使得图形处理器将以至少部分地并发的方式处理命令序列。
图形处理器命令序列2210能以管线转储清除命令2212开始,以使任何活跃的图形管线完成用于管线的当前未决的命令。任选地,3D管线2222和媒体管线2224可以不并发地操作。执行管线转储清除以使活跃的图形管线完成任何未决命令。响应于管线转储清除,用于图形处理器的命令解析器将暂停命令处理,直到活跃的绘画引擎完成未决操作并且相关的读缓存被无效。任选地,渲染缓存中被标记为“脏”的任何数据可以被转储清除到存储器。管线转储清除命令2212可用于管线同步,或可在将图形处理器置于低功率状态之前被使用。
当命令序列需要图形处理器在管线之间显式地切换时,可使用管线选择命令2213。在发出管线命令之前可在执行上下文中仅需要一次管线选择命令2213,除非上下文是发出针对这两条管线的命令。可紧接在经由管线选择命令2213进行的管线切换之前需要管线转储清除命令2212。
管线控制命令2214可配置用于操作的图形管线,并且可用于对3D管线2222和媒体管线2224进行编程。管线控制命令2214可为活跃的管线配置管线状态。管线控制命令2214可用于管线同步,并且用于在处理批量的命令之前清除来自活跃管线内的一个或多个缓存存储器的数据。
与返回缓冲器状态2216有关的命令可用于将用于相应管线的返回缓冲器的集合配置成用于写入数据。一些管线操作需要对一个或多个返回缓冲器的分配、选择或配置,在处理期间操作将中间数据写入这一个或多个返回缓冲器中。图形处理器也可使用一个或多个返回缓冲器以存储输出数据并执行跨线程通信。返回缓冲器状态2216可包括选择要用于管线操作的集合的返回缓冲器的大小和数量。
命令序列中的其余命令基于用于操作的活跃管线而不同。基于管线判定2220,命令序列被定制成用于以3D管线状态2230开始的3D管线2222、或者在媒体管线状态2240处开始的媒体管线2224。
用于配置3D管线状态2230的命令包括用于顶点缓冲器状态、顶点元素状态、常量颜色状态、深度缓冲器状态、以及将在处理3D基元命令之前配置的其他状态变量的3D状态设置命令。这些命令的值至少部分地基于使用中的特定3D API来确定。3D管线状态2230命令也可以能够在将不会使用某些管线元件的情况下选择性地禁用或绕过那些元件。
3D基元2232命令可用于提交要由3D管线处理的3D基元。经由3D基元2232命令传递给图形处理器的命令和相关联的参数被转发到图形管线中的顶点取得功能。顶点取得功能使用3D基元2232命令数据来生成顶点数据结构。顶点数据结构被存储在一个或多个返回缓冲器中。3D基元2232命令可用于经由顶点着色器对3D基元执行顶点操作。为了处理顶点着色器,3D管线2222将着色器执行线程调遣给图形处理器执行单元。
3D管线2222可经由执行2234命令或事件来触发。寄存器可写入触发命令执行。可经由命令序列中的“去往(go)”或“踢除(kick)”命令来触发执行。命令执行可使用管线同步命令以通过图形管线对命令序列转储清除来触发。3D管线将执行针对3D基元的几何处理。一旦操作完成,就对所得到的几何对象进行栅格化,并且像素引擎对所得到的像素进行着色。对于那些操作,还可以包括用于控制像素着色和像素后端操作的附加命令。
当执行媒体操作时,图形处理器命令序列2210可遵循媒体管线2224路径。一般而言,针对媒体管线2224进行编程的特定用途和方式取决于要执行的媒体或计算操作。在媒体解码期间,特定的媒体解码操作可被迁移到媒体管线。也可绕过媒体管线,并且可使用由一个或多个通用处理核心提供的资源完全地或部分地执行媒体解码。媒体管线还可包括用于通用图形处理器单元(GPGPU)操作的元件,其中,图形处理器用于使用计算着色器程序来执行SIMD向量操作,这些计算着色器程序并不显式地与图形基元的渲染相关。
能以与3D管线2222类似的方式来配置媒体管线2224。用于配置媒体管线状态2240的命令的集合在媒体对象命令2242之前被调遣或被放置到命令队列中。用于媒体管线状态2240的命令可包括用于配置将被用于处理媒体对象的媒体管线元件的数据。这包括用于在媒体管线内配置视频解码和视频编码逻辑的数据,诸如编码或解码格式。用于媒体管线状态2240的命令还可支持使用指向包含批量的状态设置的“间接”状态元素的一个或多个指针。
媒体对象命令2242可供应指向用于由媒体管线处理的媒体对象的指针。媒体对象包括存储器缓冲器,该存储器缓冲器包含要处理的视频数据。任选地,在发出媒体对象命令2242之前,所有的媒体管线状态必须是有效的。一旦管线状态被配置并且媒体对象命令2242被排队,就经由执行命令2244或等效的执行事件(例如,寄存器写入)来触发媒体管线2224。随后可通过由3D管线2222或媒体管线2224提供的操作对来自媒体管线2224的输出进行后处理。能以与媒体操作类似的方式配置和执行GPGPU操作。
图形软件体系结构
图23图示用于数据处理系统2300的示例性图形软件体系结构。此类软件体系结构可包括3D图形应用2310、操作系统2320以及至少一个处理器2330。处理器2330可包括图形处理器2332以及一个或多个通用处理器核心2334。处理器2330可以是处理器1402或本文中描述的处理器中的任何其他处理器的变体。可替代处理器1402或本文中描述的处理器中的任何其他处理器来使用处理器2330。因此,结合处理器1402或本文中描述的处理器中的任何其他处理器对任何特征的公开也公开了对应的与图形处理器2330的结合,但不限于此。此外,图23的具有与本文中任何其他附图的元件相同或类似名称的元件描述与其他附图中相同的元件,能以与其他附图中类似的方式进行操作或运行,可包括相同的部件,并且可链接到其他实体,该实体如本文中其他地方所描述的那些实体,但不限于此。图形应用2310和操作系统2320各自在数据处理系统的系统存储器2350中执行。
3D图形应用2310可包含一个或多个着色器程序,该一个或多个着色器程序包括着色器指令2312。着色器语言指令可以采用高级着色器语言,诸如,Direct3D的高级着色器语言(High-Level Shader Language,HLSL)、OpenGL着色器语言(OpenGL Shader Language,GLSL),等等。应用还可包括采用适于由通用处理器核心2334执行的机器语言的可执行指令2314。应用还可包括由顶点数据限定的图形对象2316。
操作系统2320可以是来自微软公司的操作系统、专属的类UNIX操作系统或使用Linux内核的变体的开放源类UNIX操作系统。操作系统2320可支持图形API 2322,诸如,Direct3D API、OpenGL API或Vulkan API。当Direct3D API在使用中时,操作系统2320使用前端着色器编译器2324以将采用HLSL的任何着色器指令2312编译成较低级的着色器语言。编译可以是即时(just-in-time,JIT)编译或者应用可执行着色器预编译。在3D图形应用2310的编译期间,高级着色器可被编译为低级着色器。着色器指令2312能以中间形式提供,该中间形式诸如,由Vulkan API使用的标准便携式中间表示(StandardPortable Intermediate Representation,SPIR)的某个版本。
用户模式图形驱动器2326可包含后端着色器编译器2327以将着色器指令2312编译为硬件特定表示。当OpenGL API在使用中时,将采用GLSL高级语言的着色器指令2312传递至用户模式图形驱动器2326以用于编译。用户模式图形驱动器2326可使用操作系统内核模式功能2328来与内核模式图形驱动器2329通信。内核模式图形驱动器2329可与图形处理器2332通信以调遣命令和指令。
IP核心实现方式
一个或多个方面可以由存储在机器可读介质上的代表性代码实现,该机器可读介质表示和/或限定集成电路(诸如,处理器)内的逻辑。例如,机器可读介质可包括表示处理器内的各种逻辑的指令。当由机器读取时,指令可使机器制造用于执行本文所描述的技术的逻辑。此类表示(被称为“IP核心”)是集成电路的逻辑的可重复使用单元,这些可重复使用单元可以作为描述集成电路的结构的硬件模型而被存储在有形的、机器可读介质上。可以将硬件模型供应至在制造集成电路的制造机器上加载硬件模型的各客户或制造设施。可以制造集成电路,使得电路执行与本文中描述的实施例中的任一实施例相关联地描述的操作。
图24A是图示根据实施例的可用于制造集成电路以执行操作的IP核心开发系统2400的框图。IP核心开发系统2400可以用于生成可并入到更大的设计中或用于构建整个集成电路(例如,SOC集成电路)的模块化、可重复使用的设计。设计设施2430可生成采用高级编程语言(例如,C/C++)的IP核心设计的软件仿真2410。软件仿真2410可用于使用仿真模型2412来设计、测试并验证IP核心的行为。仿真模型2412可包括功能仿真、行为仿真和/或时序仿真。随后可从仿真模型2412创建或合成寄存器传输级(register transfer level,RTL)设计2415。RTL设计2415是对硬件寄存器之间的数字信号的流进行建模的集成电路(包括使用建模的数字信号来执行的相关联的逻辑)的行为的抽象。除了RTL设计2415之外,还可创建、设计或合成逻辑级或晶体管级的较低级别设计。由此,初始设计和仿真的特定细节可有所不同。
可由设计设施进一步将RTL设计2415或等效方案合成到硬件模型2420中,该硬件模型2420可以采用硬件描述语言(hardware description language,HDL)或物理设计数据的某种其他表示。可以进一步仿真或测试HDL以验证IP核心设计。可使用非易失性存储器2440(例如,硬盘、闪存或任何非易失性存储介质)来存储IP核心设计以用于递送至第三方制造设施2465。替代地,可通过有线连接2450或无线连接2460(例如,经由互联网)来传送IP核心设计。制造设施2465随后可制造至少部分地基于IP核心设计的集成电路。所制造的集成电路可被配置成用于执行根据本文中描述的至少一个实施例的操作。
图24B图示集成电路封装组件2470的截面侧视图。集成电路封装组件2470图示如本文中所描述的一个或多个处理器或加速器设备的实现方式。封装组件2470包括连接至衬底2480的多个硬件逻辑单元2472、2474。逻辑2472、2474可至少部分地在可配置逻辑或固定功能逻辑硬件中实现,并且可包括本文中描述的(一个或多个)处理器核心、(一个或多个)图形处理器或其他加速器设备中的任一者的一个或多个部分。每个逻辑单元2472、2474可在半导体管芯内实现,并且经由互连组织2473与衬底2480耦合。互连组织2473可被配置成用于在逻辑2472、2474与衬底2480之间路由电信号,并且可包括互连,该互连诸如但不限于凸块或支柱。互连组织2473可被配置成路由电信号,诸如例如,与逻辑2472、2474的操作相关联的输入/输出(I/O)信号和/或功率或接地信号。任选地,衬底2480可以是基于环氧树脂的层压衬底。衬底2480还可包括其他合适类型的衬底。封装组件2470可经由封装互连2483连接到其他电气设备。封装互连2483可耦合至衬底2480的表面以将电信号路由到其他电气设备,诸如主板、其他芯片组或多芯片模块。
逻辑单元2472、2474可与桥接器2482电耦合,该桥接器2482被配置成用于在逻辑2472与逻辑2474之间路由电信号。桥接器2482可以是为电信号提供路由的密集互连组织。桥接器2482可包括由玻璃或合适的半导体材料构成的桥接器衬底。电路由特征可形成在桥接器衬底上,以提供逻辑2472与逻辑2474之间的芯片到芯片连接。
尽管图示了两个逻辑单元2472、2474和桥接器2482,但是本文中所描述的实施例可包括在一个或多个管芯上的更多或更少的逻辑单元。这一个或多个管芯可以由零个或更多个桥接器连接,因为当逻辑被包括在单个管芯上时,可以排除桥接器2482。替代地,多个管芯或逻辑单元可以由一个或多个桥接器连接。此外,多个逻辑单元、管芯和桥接器可按其他可能的配置(包括三维配置)被连接在一起。
图24C图示封装组件2490,该封装组件2490包括连接到衬底2480(例如,基础管芯)的多个单元的硬件逻辑小芯片。如本文中所描述的图形处理单元、并行处理器和/或计算加速器可由分开制造的各种硅小芯片组成。在该上下文中,小芯片是至少部分地被封装的集成电路,该至少部分地被封装的集成电路包括可与其他小芯片一起被组装到更大的封装中的不同的逻辑单元。具有不同IP核心逻辑的各种集合的小芯片可被组装到单个设备中。此外,小芯片可使用有源中介层(interposer)技术而被集成到基础管芯或基础小芯片中。本文中描述的概念启用GPU内的不同形式的IP之间的互连和通信。IP核心可使用不同的工艺技术来制造并在制造期间被构成,这避免了尤其是对于具有若干风格的IP的大型SoC的将多个IP汇聚到同一制造工艺的复杂性。允许使用多种工艺技术改善了上市时间,并提供具有成本效益的方法来创建多个产品SKU。此外,分解的IP更易修改以被独立地功率门控,对于给定工作负载不在使用中的部件可被关断,从而降低总功耗。
在各实施例中,封装组件2490可包括由结构2485或一个或多个桥接器2487互连的更少或更多数量的部件和小芯片。封装组件2490内的小芯片可具有使用芯片-晶片-衬底(Chip-on-Wafer-on-Substrate)堆叠的2.5D布置,其中,多个管芯并排地堆叠在硅中介层上,该硅中介层包括硅通孔(through-silicon vias,TSV)以将小芯片与衬底2480耦合,该衬底2480包括至封装互连2483的电气连接。
在一个实施例中,硅中介层是有源中介层2489,该有源中介层2489除了TSV之外还包括嵌入式逻辑。在此类实施例中,封装组件2490内的小芯片使用3D面对面管芯堆叠被布置在有源中介层2489的顶部上。有源中介层2489除互连结构2485和硅桥接器2487外还可包括用于I/O2491的硬件逻辑、缓存存储器2492和其他硬件逻辑2493。结构2485启用各种逻辑小芯片2472、2474与有源中介层2489内的逻辑2491、2493之间的通信。结构2485可以是在封装组件的部件之间交换数据分组的NoC互连或另一形式的分组交换型结构。对于复杂组件,结构2485可以是启用封装组件2490的各硬件逻辑之间的通信的专用小芯片。
有源中介层2489内的桥接器结构2487可用于促进例如逻辑或I/O小芯片2474与存储器小芯片2475之间的点到点互连。在一些实现方式中,桥接器结构2487还可被嵌入在衬底2480内。
硬件逻辑小芯片可包括专用硬件逻辑小芯片2472、逻辑或I/O小芯片2474和/或存储器小芯片2475。硬件逻辑小芯片2472以及逻辑或I/O小芯片2474可以至少部分地在可配置逻辑或固定功能逻辑硬件中实现,并且可包括本文中描述的(一个或多个)处理器核心、(一个或多个)图形处理器、并行处理器或其他加速器设备中的任一个的一个或多个部分。存储器小芯片2475可以是DRAM(例如,GDDR、HBM)存储器或缓存(SRAM)存储器。有源中介层2489(或衬底2480)内的缓存存储器2492可充当用于封装组件2490的全局缓存,充当分布式全局缓存的部分,或充当用于结构2485的专用缓存。
每个小芯片可被制造为单独的半导体管芯,并且可与基础管芯耦合,该基础管芯嵌入在衬底2480内或与衬底2480耦合。与衬底2480的耦合可经由互连组织2473来执行。互连组织2473可被配置成用于在衬底2480内的各种小芯片与逻辑之间路由电信号。互连组织2473可包括互连,诸如但不限于凸块或支柱。在一些实施例中,互连组织2473可被配置成用于路由电信号,诸如例如,与逻辑、I/O和存储器小芯片的操作相关联的输入/输出(I/O)信号和/或功率或接地信号。在一个实施例中,附加的互连组织将有源中介层2489与衬底2480耦合。
衬底2480可以是基于环氧树脂的层压衬底,然而,它不限于此,并且衬底2480还可包括其他合适类型的衬底。封装组件2490可经由封装互连2483连接到其他电气设备。封装互连2483可耦合至衬底2480的表面以将电信号路由到其他电气设备,诸如,主板、其他芯片组或多芯片模块。
逻辑或I/O小芯片2474和存储器小芯片2475可经由桥接器2487被电耦合,该桥接器2487被配置成用于在逻辑或I/O小芯片2474与存储器小芯片2475之间路由电信号。桥接器2487可以是为电信号提供路由的密集互连组织。桥接器2487可包括由玻璃或合适的半导体材料构成的桥接器衬底。电路由特征可形成在桥接器衬底上以提供逻辑或I/O小芯片2474与存储器小芯片2475之间的芯片到芯片连接。桥接器2487还可被称为硅桥接器或互连桥接器。例如,桥接器2487是嵌入式多管芯互连桥接器(Embedded Multi-dieInterconnect Bridge,EMIB)。替代地,桥接器2487可简单地是从一个小芯片到另一小芯片的直接连接。
图24D图示根据实施例的包括可互换小芯片2495的封装组件2494。可互换小芯片2495可被组装到一个或多个基础小芯片2496、2498上的标准化插槽中。基础小芯片2496、2498可经由桥接器互连2497被耦合,该桥接器互连2497可与本文中描述的其他桥接器互连类似,并且可以是例如EMIB。存储器小芯片也可经由桥接器互连被连接到逻辑或I/O小芯片。I/O和逻辑小芯片可经由互连结构进行通信。基础小芯片各自都能以用于逻辑或I/O或存储器/缓存中的一者的标准化格式来支持一个或多个插槽。
SRAM和功率递送电路可被制造到基础小芯片2496、2498中的一个或多个中,基础小芯片2496、2498可使用相对于可互换小芯片2495不同的工艺技术来制造,可互换小芯片2495堆叠在基础小芯片的顶部上。例如,可使用较大工艺技术来制造基础小芯片2496、2498,同时可使用较小工艺技术来制造可互换小芯片。可互换小芯片2495中的一个或多个可以是存储器(例如,DRAM)小芯片。可基于针对使用封装组件2494的产品的功率和/或性能来为封装组件2494选择不同的存储器密度。此外,可在组装时基于针对产品的功率和/或性能来选择具有不同数量的类型的功能单元的逻辑小芯片。此外,可将包含具有不同类型的IP逻辑核心的小芯片插入到可互换小芯片插槽中,从而启用可混合并匹配不同技术的IP块的混合式处理器设计。
示例性片上系统集成电路
图25-图26B图示可使用一个或多个IP核心制造的示例性集成电路和相关联的图形处理器。除了所图示的内容之外,还可包括其他逻辑和电路,包括附加的图形处理器/核心、外围接口控制器或通用处理器核心。图25-图26B的具有与本文中任何其他附图的元件相同或类似名称的元件描述与其他附图中相同的元件,能以与其他附图中类似的方式进行操作或运行,可包括相同的部件,并且可链接到其他实体,该实体如本文中其他地方所描述的那些实体,但不限于此。
图25是图示可使用一个或多个IP核心来制造的示例性片上系统集成电路2500的框图。示例性集成电路2500包括一个或多个应用处理器2505(例如,CPU)、至少一个图形处理器2510,该至少一个图形处理器2510可以是图形处理器1408、1508、2510的变体,或者可以是本文中描述的并且可替代所描述的任何图形处理器被使用的任何图形处理器。因此,本文中结合图形处理器对任何特征的公开也公开了对应的与图形处理器2510的结合,但不限于此。集成电路2500可附加地包括图像处理器2515和/或视频处理器2520,图像处理器2515和视频处理器2520中的任一者可以是来自相同的设计设施或多个不同的设计设施的模块化IP核心。集成电路2500可包括外围或总线逻辑,包括USB控制器2525、UART控制器2530、SPI/SDIO控制器2535和I2S/I2C控制器2540。此外,集成电路可包括显示设备2545,该显示设备2545耦合至高清晰度多媒体接口(high-definition multimedia interface,HDMI)控制器2550和移动行业处理器接口(mobile industry processor interface,MIPI)显示接口2555中的一个或多个。可以由闪存子系统2560(包括闪存和闪存控制器)来提供存储。可以经由存储器控制器2565来提供存储器接口以获得对SDRAM或SRAM存储器设备的访问。一些集成电路附加地包括嵌入式安全引擎2570。
图26A-图26B是图示根据本文中所描述的实施例的用于在SoC内使用的示例性图形处理器的框图。所图示的图形处理器可以是图形处理器1408、1508、2510、或本文中描述的任何其他图形处理器的变体。图形处理器可替代图形处理器1408、1508、2510、或本文中描述的图形处理器中的任何其他图形处理器被使用。因此,结合图形处理器1408、1508、2510或本文中描述的图形处理器中的任何其他图形处理器对任何特征的公开也公开了对应的与图26A-图26B的图形处理器的结合,但不限于此。图26A图示根据实施例的可以使用一个或多个IP核心来制造的片上系统集成电路的示例性图形处理器2610。图26B图示根据实施例的可以使用一个或多个IP核心来制造的片上系统集成电路的附加的示例性图形处理器2640。图26A的图形处理器2610是低功率图形处理器核心的示例。图26B的图形处理器2640是较高性能的图形处理器核心的示例。例如,如本段开头所提及,图形处理器2610和图形处理器2640中的每个图形处理器可以是图25的图形处理器2510的变体。
如图26A中所示,图形处理器2610包括顶点处理器2605和一个或多个片段处理器2615A-2615N(例如,2615A、2615B、2615C、2615D,一直到2615N-1和2615N)。图形处理器2610可以经由单独的逻辑执行不同的着色器程序,使得顶点处理器2605被优化以执行用于顶点着色器程序的操作,而一个或多个片段处理器2615A-2615N执行用于片段或像素着色器程序的片段(例如,像素)着色操作。顶点处理器2605执行3D图形管线的顶点处理阶段,并生成基元和顶点数据。(一个或多个)片段处理器2615A-2615N使用由顶点处理器2605生成的基元数据和顶点数据来产生被显示在显示设备上的帧缓冲器。(一个或多个)片段处理器2615A-2615N可被优化以执行如在OpenGL API中提供的片段着色器程序,这些片段着色器程序可以用于执行与如在Direct 3D API中提供的像素着色器程序类似的操作。
图形处理器2610附加地包括一个或多个存储器管理单元(MMU)2620A-2620B、(一个或多个)缓存2625A-2625B以及(一个或多个)电路互连2630A-2630B。该一个或多个MMU2620A-2620B为图形处理器2610(包括为顶点处理器2605和/或(一个或多个)片段处理器2615A-2615N)提供虚拟到物理地址映射,除了存储在一个或多个缓存2625A-2625B中的顶点数据或图像/纹理数据之外,该虚拟到物理地址映射还可以引用存储在存储器中的顶点数据或图像/纹理数据。一个或多个MMU 2620A-2620B可以与系统内的其他MMU同步,使得每个处理器2505-2520可以参与共享或统一的虚拟存储器系统,系统内的其他MMU包括与图25的一个或多个应用处理器2505、图像处理器2515和/或视频处理器2520相关联的一个或多个MMU。图形处理器2610的部件可与本文中描述的其他图形处理器的部件相对应。一个或多个MMU 2620A-2620B可与图2C的MMU 245相对应。顶点处理器2605和片段处理器2615A-2515N可与图形多处理器234相对应。根据实施例,一个或多个电路互连2630A-2630B使得图形处理器2610能够经由SoC的内部总线或经由直接连接来与SoC内的其他IP核心对接。一个或多个电路互连2630A-2630B可与图2C的数据交叉开关240相对应。可在图形处理器2610的类似部件与本文中描述的各种图形处理器体系结构之间发现进一步的对应性。
如图26B中所示,图形处理器2640包括图26A的图形处理器2610的一个或多个MMU2620A-2620B、缓存2625A-2625B和电路互连2630A-2630B。图形处理器2640包括一个或多个着色器核心2655A-2655N(例如,2655A、2655B、26555C、2655D、2655E、2655F,一直到2655N-1和2655N),其提供统一的着色器核心体系结构,其中,单个核心或类型或核心可执行所有类型的可编程着色器代码,包括用于实现顶点着色器、片段着色器和/或计算着色器的着色器程序代码。存在的着色器核心的确切数量可以因实施例和实现方式而异。此外,图形处理器2640包括核心间任务管理器2645,该核心间任务管理器2645充当用于将执行线程调遣给一个或多个着色器核心2655A-2655N的线程调遣器和用于加速对基于片的渲染的分片操作的分片单元2658,在基于片的渲染中,针对场景的渲染操作在图像空间中被细分,例如以利用场景内的局部空间一致性或优化内部缓存的使用。着色器核心2655A-2655N可例如与图2D中的图形多处理器234相对应,或分别与图3A和图3B的图形多处理器325、350相对应,或与图3C的多核心组365A相对应。
用于图形和机器学习工作负载的张量加速逻辑
图27是根据实施例的数据处理系统2700的框图。数据处理系统2700是异构处理系统,其具有处理器2702、统一存储器2710和包括机器学习加速逻辑的GPGPU 2720。处理器2702和GPGPU 2720可以是如本文中所描述的处理器和GPGPU/并行处理器中的任一者。例如,另外参考图1,处理器2702可以是所图示的一个或多个处理器102中的处理器的变体和/或与所图示的一个或多个处理器102中的处理器共享体系结构,并且GPGPU 2720可以是所图示的一个或多个并行处理器112中的并行处理器的变体和/或与所图示的一个或多个并行处理器112中的并行处理器共享体系结构。另外参考图14,处理器2702可以是(一个或多个)所图示的处理器1402中的一个处理器的变体和/或与(一个或多个)所图示的处理器1402中的一个处理器共享体系结构,并且GPGPU 2720可以是(一个或多个)所图示的图形处理器1408中的一个图形处理器的变体和/或与(一个或多个)所图示的图形处理器1408中的一个图形处理器共享体系结构。
处理器2702可执行存储在系统存储器2712中的用于编译器2715的指令。编译器2715在处理器2702上执行,以将源代码2714A编译成编译代码2714B。编译代码2714B可包括可由处理器2702执行的指令和/或可由GPGPU 2720执行的指令。要由GPGPU执行的指令的编译可使用着色器或计算程序编译器(诸如,图23中的着色器编译器2327和/或着色器编译器2324)促进。在编译期间,编译器2715可执行操作以插入元数据,该元数据包括关于编译代码2714B中存在的数据并行性水平的提示和/或关于与要基于编译代码2714B而被调遣的线程相关联的数据局部性的提示。编译器2715可包括对于执行此类操作必要的信息,或者可以在运行时库2716的辅助下执行这些操作。运行时库2716还可在源代码2714A的编译时辅助编译器2715,并且还可包括指令,这些指令在运行时与编译代码2714B链接以促进编译指令在GPGPU 2720上的执行。编译器2715还可促进经由寄存器分配器(register allocator,RA)对变量的寄存器分配,并且生成用于在存储器与为变量指派的寄存器之间移动用于该变量的数据的加载和存储指令。
统一存储器2710表示可由处理器2702和GPGPU 2720访问的统一地址空间。统一存储器可包括系统存储器2712以及GPGPU存储器2718。GPGPU存储器2718是GPGPU 2720的地址空间内的存储器,并且可包括系统存储器2712中的一些或全部。在一个实施例中,存储在系统存储器2712中的编译代码2714B可被映射到GPGPU存储器2718中,以供由GPGPU 2720访问。GPGPU存储器2718还包括GPGPU 2720的GPGPU本地存储器2728。GPGPU本地存储器2728可包括例如HBM或GDDR存储器。
GPGPU 2720包括多个计算块2724A-2724N,这些计算块2724A-2724N可包括本文中描述的各种处理资源中的一种或多种。处理资源可以是或可包括各种不同的计算资源,诸如例如,执行单元、计算单元、流式多处理器、图形多处理器或多核心组。在一个实施例中,GPGPU 2720附加地包括张量加速器2723(例如,矩阵加速器),该张量加速器2723可包括被设计为用于加速矩阵操作的子集(例如,点积等)的一个或多个专用计算单元。张量加速器2723也可以被称为张量加速器或张量核心。在一个实施例中,张量加速器2723内的逻辑部件可跨多个计算块2724A-2724N的处理资源而分布。在一个实施例中,GPGPU 2720附加地包括光流生成器(OFG 2729),其可配置成用于确定两个或更多个输入帧之间的前向或后向光流。OFG 2729可以是专用光流加速器,其生成处于或接近输入帧的分辨率的高分辨率光流数据,或者是视频编解码器电路内的多用途光流分析器。总体上,编解码器生成的光流数据的分辨率小于帧的像素数据的分辨率,但在功耗和面积消耗方面更有效,并且可以出于帧生成以外的目的而生成光流数据。在一个实施例中,OFG 2729可通过软件配置来生成高分辨率光流数据或较低分辨率光流以满足实时等待时间需求。
GPGPU 2720还可包括可由计算块2724A-2724N和张量加速器2723共享的资源的集合,该资源的集合包括但不限于寄存器的集合2725、功率和性能模块2726、和缓存2727。在一个实施例中,寄存器2725包括直接可访问和间接可访问的寄存器,其中间接可访问的寄存器被优化以供由张量加速器2723使用。功率和性能模块2726可被配置成用于调整用于计算块2724A-2724N的功率递送和时钟频率,以对计算块2724A-2724N内的空闲部件进行功率门控。在各个实施例中,缓存2727可包括指令缓存和/或低级数据缓存。
GPGPU 2720可附加地包括L3数据缓存2730,该L3数据缓存2730可用于对由张量加速器2723和/或计算块2724A-2724N内的计算元件从统一存储器2710访问的数据进行缓存。在一个实施例中,L3数据缓存2730包括共享本地存储器2732,该共享本地存储器2732可由计算块2724A-2724N内的计算元件和张量加速器2723共享。
在一个实施例中,GPGPU 2720包括指令处置逻辑,诸如,取得(fetch)和解码单元2721以及调度器控制器2722。取得和解码单元2721包括取得单元和解码单元,用于取得指令并对指令解码以供由计算块2724A-2724N中的一个或多个或由张量加速器2723执行。指令可经由调度器控制器2722被调度给计算块2724A-2724N或张量加速器内的适当的功能单元。在一个实施例中,调度器控制器2722是能够配置成用于执行高级调度操作的ASIC。在一个实施例中,调度器控制器2722是能够执行从固件模块加载的调度器指令的微控制器或低每指令能量的处理核心。
在一个实施例中,用于由计算块2724A-2724N执行的一些功能可以被直接调度或转移(offload)到张量加速器2723。在各种实施例中,张量加速器2723包括被配置成用于高效地执行矩阵计算操作的处理元件逻辑,矩阵计算操作诸如由3D图形或计算着色器程序使用的乘法和加法操作以及点积操作。在一个实施例中,张量加速器2723可被配置成用于加速由机器学习框架使用的操作。在一个实施例中,张量加速器2723是显式地被配置成用于执行并行矩阵乘法和/或加法操作的特定集合的专用集成电路。在一个实施例中,张量加速器2723是现场可编程门阵列(FPGA),其提供可在工作负载之间被更新的固定功能逻辑。在一个实施例中,可由张量加速器2723执行的计算操作的集合相对于可由计算块2724A-2724N执行的操作可能是有限的。然而,张量加速器2723能够以相对于计算块2724A-2724N显著更高的吞吐量执行并行张量操作。
图28A-图28B图示根据实施例的由指令管线2800执行的矩阵操作2805。图28A图示当配置有张量加速器2723内的脉动阵列2808时的指令管线2800。图28图示当配置有各自都包括矩阵引擎2812A-2812N的图形处理器核心2810A-2810N时的指令管线。
如图28A中所示,指令管线2800可被配置成用于执行矩阵操作2805,诸如但不限于点积操作。两个向量的点积是等于向量的对应分量的乘积之和的标量值。可以如以下等式(1)中所示来计算点积。
点积可在用于卷积神经网络(convolutional neural network,CNN)的卷积操作中使用。尽管图示2D卷积,但是可以使用N维滤波器对N维体积执行N维卷积。感受野片2802突出显示输入体积缓冲器2804中的输入体积的部分。输入体积缓冲器可以被存储在存储器2830中。可以在感受野片2802内的数据与卷积滤波器内的数据之间执行点积矩阵操作2805以在输出缓冲器2806内生成数据点,该数据点也可以被存储在存储器2830中。存储器2830可以是本文中描述的存储器中的任一个,包括如图27中的系统存储器2712、GPGPU存储器2718或一个或多个缓存存储器2727、2730。
输出缓冲器2806内的数据点的组合表示由卷积操作生成的激活图。激活图内的每个点通过将感受野片滑过输入体积缓冲器2804来生成。可以将激活图数据输入到激活函数以确定输出激活值。在一个实施例中,可以在框架内将输入体积缓冲器2804的卷积定义为高级矩阵操作2805。可以经由基元操作(诸如,基本线性代数子程序(basic linearalgebra subprogram,BLAS)操作)来执行高级矩阵操作。可以经由指令管线2800执行的硬件指令来加速基元操作。
用于加速硬件指令的指令管线2800可包括:指令取得和解码单元2721,其可取得硬件指令并对硬件指令解码;以及调度器控制器2722,其可将经解码的指令调度到计算块2724A-2724N和/或张量加速器2723内的一个或多个处理资源。在一个实施例中,硬件指令可被调度到计算块2724A-2724N,并且可被迁移到张量加速器2723。用于执行矩阵操作2805的一个或多个硬件指令和相关联的数据可以被存储在存储器2830中。硬件指令的输出也可以被存储在存储器2830中。
在一个实施例中,张量加速器2723可执行一个或多个硬件指令以使用处理元件的脉动阵列2808来执行矩阵操作2805。脉动阵列2808包括能够被配置成用于执行矩阵-矩阵点积操作和矩阵-向量点积操作以及其他操作(诸如,矩阵-矩阵融合乘加操作和矩阵-向量融合乘加操作)的可编程硬件和固定功能硬件的组合。脉动阵列2808包括矩阵引擎的阵列,其中每个矩阵引擎可以类似于图18C的矩阵引擎1803来配置。
在各实施例中,作为张量加速器2723的替代或附加,矩阵加速逻辑也可被包括在计算块2724A-2724N的处理资源内。例如,如图28B中所示,在一个实施例中,每个计算块(例如,计算块2724N)包括图形核心2810A-2810N的阵列。图形核心2810A-2810N的阵列中的每个图形核心可包括矩阵加速器2812A-2812N。在一个实施例中,图形核心2810A-2810N是图18A中的图形核心1815A-1815N,并且矩阵加速器2812A-2812N包括图18C的矩阵引擎1803的某个版本。调度器控制器2722可将矩阵操作(点积、融合乘加等)调度到各种计算块2724A-2724N的图形核心2810A-2810N内的可用的矩阵加速器2812A-2812N。
尽管在一个实施例中计算块2724A-2724N中的每个计算块包括图形核心2810A-2810N的阵列,但是在另一实施例中,计算块2724A-2724N与图2A中的处理集群阵列的处理集群214A-214N共享体系结构。在此类实施例中,计算块2724A-2724N包括图2C中的多个图形多处理器234,多个图形多处理器234包括如图2D中图示的内部部件。由此,计算块内的图形多处理器可包括加载/存储单元266、GPGPU核心262、和/或张量/RT核心263。在一个实施例中,计算块2724A-2724N可包括图3C的GPU 380的多核心组365A-365N,并且可包括多个集合的GFX(图形)核心370、张量核心371和光线追踪核心372。在此类实施例中,调度器控制器2722可将用于执行矩阵操作的指令调度到计算块2724A-2724N内的张量/RT核心263和/或张量核心371。经加速的矩阵操作包括点积操作、矩阵乘法操作和/或融合乘加操作,这些操作可对整数或浮点矩阵元素以及各种精度级别执行。附加地,在一个实施例中,计算块2724A-2724N可包括图15C的计算单元1560A-1560N的变体,其中,此类变体包括能够执行整数或浮点矩阵加速指令的如本文中所描述的矩阵加速逻辑(例如,脉动阵列、张量核心、脉动张量核心)。在每个配置中,每个计算块2724A-2724N内的处理元件可以协作来执行内核程序的线程块集群。
图29图示计算块2900,该计算块2900包括启用编解码器的分解的脉动逻辑。在一个实施例中,与在图28A中在单独的张量加速器2723中包括脉动阵列2808、或者在每个图形核心2815A-2815N中包括矩阵引擎2812A-2812N不同,脉动阵列的分解的集合2912A-2912B可以被包括在计算块2900中,该计算块2900与图27的计算块2724A-2724N中的一个类似。计算块2900可包括多个互连的处理资源(PR 2908A-2908O),它们可与本文中描述的任何处理资源体系结构类似,诸如但不限于本文中描述的处理资源。在一个实施例中,脉动阵列2912A-2912B包括编解码器2924A-2924B,该编解码器2924A-2924B使得能够对被接收以进行处理的输入数据和输出数据编码和解码。
脉动阵列2912A-2912B包括数据处理单元的宽W且深D的网络,其可用于以脉动方式执行向量或其他数据并行操作,与本文中描述的其他脉动阵列类似。在一个实施例中,脉动阵列2912A-2912B可被配置成用于执行矩阵操作,诸如,矩阵点积操作。在一个实施例中,脉动阵列2912A-2912B支持16比特和8比特的浮点操作,以及8比特和4比特整数操作、三元操作、二进制、双极二进制、三元和独热(one-hot)比特操作。在一个实施例中,脉动阵列2912A-2912B可被配置成用于加速机器学习操作。在此类实施例中,脉动阵列2912A-2912B可被配置有对brain浮点(bfloat)16比特浮点格式的支持。通过在计算块2900内但是在PR2908A-2908O外部包括脉动阵列2912A-2912B,脉动阵列2912A-2912B的大小和数量可以独立于PR 2908A-2908O的数量而缩放。附加地,可以保留PR内的原本将由脉动阵列活动消耗的通信带宽。此外,当矩阵工作负载不是正在被执行时,脉动阵列2912A-2912B可以被时钟/功率门控。
脉动阵列2912A-2912B和PR 2908A-2908O之间的通信可以经由缓存或共享本地存储器(缓存/SLM 2910)和/或共享寄存器堆2914来执行。在一个实施例中,代替分立的共享寄存器堆2914,可以对缓存/SLM2910分区以用作共享寄存器堆。共享寄存器堆2914可类似地被结构化到本文中描述的其他GPGPU寄存器堆。共享寄存器堆还可以包括用于配置脉动阵列2912A-2912B与PR 2908A-2908O之间的交互的专用寄存器的集合。缓存/SLM 2910可以是L1缓存、L2缓存和/或显式地可寻址的管芯上存储器的块。
用于由脉动阵列2912A-2912B处理的矩阵数据可以被存储在缓存/SLM 2910中。处理命令或指令可以经由共享寄存器堆2914被提供给脉动阵列2912A-2912B。处理结果可以由PR 2908A-2908O从缓存/SLM 2910读取或者从共享寄存器堆内的目的地/输出寄存器读取。在操作期间,通信流量可以本地化到脉动阵列2912A-2912B、缓存/SLM 2910和/或共享寄存器堆2914,而不是消耗PR 2908A-2908O内的总线/结构带宽。计算块2900内的PR2908A-2908O中的任一个可以将矩阵工作负载迁移到脉动阵列2912A-2912B中的一个或两个。可以将具有命令的消息从PR发送至脉动阵列,该命令指定要执行的操作和用于该操作的操作对象。脉动阵列2912A-2912B可以执行所请求的操作(乘法/加法、融合乘法/加法、乘法/累加、点积等)并且将结果输出到共享寄存器堆2914。用于所请求的操作的输入数据、中间数据和/或输出数据可以被存储在缓存/SLM 2910中,并且多个依赖性操作可以被链接。在一个实施例中,当用于神经网络的训练或推断的处理操作被执行时,脉动阵列2928A-2928B还可以执行激活函数,这些激活函数包括但不限于sigmoid激活、ReLU激活、和双曲正切(TanH)激活。在此类实施例中,用于神经网络的操作可以以粗粒度被转移至脉动阵列2912A-2912B。
PR 2908A-2908O可以以压缩格式将输入数据提供到脉动阵列2912A-2912B,并且编解码器2924A-2924B可用于对数据解压缩。当输出数据准备好提供到PR 2908A-2908O时,如果PR将执行操作和数据并且不支持直接读取经压缩的数据,则数据可以保持解压缩的状态。如果PR2908A-2908O支持读取经压缩的数据或将不会对数据执行附加操作,则输出数据可能会被重新编码。可以使用基于零的编码,并且可以基于数据稀疏度的程度来启用或禁用压缩。替代地,可以基于要被处理或输出的数据集的分布来使用其他形式的编码。例如,编解码器2924A-2924B可被配置成用于对稀疏数据解码,该稀疏数据根据基于零的压缩或者使用本文中描述的另一形式的压缩(例如,基于一、基于二、接近零、接近一、接近二等)被编码。可以被支持的附加的示例性编码或压缩技术包括唯一绝对值(unique absolutevalue,UAV)表编码、重要性图谱(significance map,SM)编码、表编码(table encodeing,TE)、唯一值坐标(unique value coordinate,UVC)编码、和均值编码(mean encoding,ME)。用于经编码数据的元数据指示用于数据的编码格式的类型。在一个实施例中,可以为特定类型的数据选择特定编码格式,特定类型的数据诸如核(kernel)数据或特征数据。在一个实施例中,在编码之前对数据执行统计分析以使得能够为每个数据块选择适当的编码器。在一个实施例中,在SM编码期间生成的数据可用于促进将经压缩的数据提供给脉动阵列2912A-2912B。在基于零的SM编码模式中,仅块中的非零值被编码。在头部中指示样本块中的非零值的数量,接着是指示块内的非零值的图谱的重要性图谱。然后按照在流内出现的顺序来对样本的非零值编码。
用于帧内插/外推的等待时间感知统一神经网络
针对渲染应用的现有帧生成技术(例如,深度学习超级采样(Deep LearningSuper Sampling,DLSS)3等)使用帧内插技术来生成中间帧,从而以增加的等待时间为代价来增加有效帧率。基于外推的帧生成技术总体上是数据密集型的,并且在计算上禁止以更高的分辨率(例如,4K)实时使用。附加地,现有技术(诸如ExtraNet)可能不适于云到客户端的用例,在该用例中,丢弃的帧由于执行外推所需的发送多个缓冲区的增加的存储器和复杂性而由客户端外推。PSUNet要求更少的输入,但在计算上是禁止的,并且此外,PSUNet要求对低于置信阈值的部分的重新渲染。
本文中描述的是统一内插/外推网络,该网络的大小小于现有网络的大小,要求更少的输入,并且易于重新配置以用于外推和内插。附加地,该网络实现了每像素零采样(0spp)帧生成,其不要求部分帧渲染或对低于置信阈值的部分的重新渲染。因此,该网络适于各种用例,这些用例包括云到客户端用例,在这些用例中,当前最先进的内插和外推方法并不适合。
统一网络的基本网络体系结构和拓扑针对内插和外推是相同的。对网络唯一要求的改变是输入帧的顺序和用于帧生成的预训练的权重。统一网络利用具有修改的U-Net体系结构的单个神经网络,该单个神经网络接受两个经渲染的帧和那些帧之间的光流作为输入。与本领域已知的其他网络相比,相对较小的大小和减少的输入数量为独立软件供应商将帧生成集成到其应用中施加了较低的集成负担,这使得帧生成能够容易地被采用。
在一个实施例中,神经网络包括具有多个卷积阶段的编码器部分、具有多个上卷积(转置卷积)阶段的解码器部分、以及最终去噪阶段。解码器部分还包括与相关联的上卷积阶段成对的附加卷积阶段。这些成对的解码器卷积阶段用于生成处于各种中间分辨率的中间光流估计,这些中间光流估计与经由编码器阶段与解码器阶段之间的跳跃连接提供的它们相关联的上卷积阶段和中间解码器阶段数据的输出串接。在训练期间,中间光流估计可以被上采样到输入分辨率,并用于扭曲渲染帧以生成预测帧。然后,经由针对网络实现中间监督的损失函数将该预测帧与基准真值(ground truth)渲染帧进行比较。最终去噪阶段输出最终过滤的光流估计,该估计使得能够通过扭曲输入渲染帧来生成新的内插帧或外推帧。
可以使用例如图28A中的张量加速器2723和相关联的脉动阵列2808、如图28B中的图形处理器核心2810A-2810N内的矩阵引擎2812A-2812N、如图29中的脉动阵列2912A-2912B来加速用于统一神经网络的神经网络操作。附加地,可以使用本文中描述的任何矩阵或张量加速器(诸如本文中描述的张量核心(例如,张量核心337A-337B)或矩阵引擎1803A-1803N)来加速神经网络操作。在一些配置中,操作的子集可以经由用于不具有矩阵加速能力或具有有限的矩阵加速能力的图形处理器的向量引擎1802A-1802N来分解以获得性能,但是适当的操作可能针对此类配置而被限制于较低的源和目标分辨率。
图30A-图30B图示根据实施例的、用于使用统一机器学习模型进行帧外推和帧内插的系统。图30A图示用于帧外推的系统3000。图30B图示用于帧内插的系统3010。一般来说,帧外推被认为是相对于帧内插而言更难的问题,因为它根据过去的帧以及与这些过去的帧相关的其他输入(诸如它们之间的光流)来预测未来的帧。相比之下,帧内插是相对容易的问题,因为它从过去的帧和当前的帧预测中间帧。然而,帧内插的等待时间成本较高,并且可能不适于某些用例。通过启用帧内插和外推两者,用户可以选择针对具有较高的质量要求的等待时间不敏感的应用使用帧内插,并且针对对等待时间敏感但对感知质量具有相对不太严格的要求的应用使用帧外推。
如图30A中所示,被配置成用于帧外推的系统3000可以用外推权重3007来加载统一神经网络3005,外推权重3007通过训练统一神经网络3005基于两个现有帧来外推未来的帧而生成。第一帧3001(帧1)和第二帧3002(帧2)可以连同帧1与帧2之间的光流3006一起被提供作为至统一神经网络3005的输入。然后,统一神经网络3005可以生成外推帧3003(帧3)。
如图30B中所示,被配置成用于帧内插的系统3010可以用内插权重3017来加载统一神经网络3005,内插权重3017通过训练统一神经网络3005基于两个现有帧来对中间帧进行内插而生成。第一帧3011(帧1)和第三帧3013(帧3)可以连同帧1与帧3之间的光流3016一起被提供作为至统一神经网络3005的输入。然后,统一神经网络3005可以在帧1和帧3之间生成内插帧3012(帧2)。
图31A-图31B图示根据实施例的、用于使用统一机器学习模型进行帧外推和帧内插的方法。图31A图示用于执行帧外推以基于先前渲染帧来预测未来帧的方法3100。图31B图示用于执行帧内插以在两个渲染帧之间生成中间帧的方法3110。
如图31A中所示,被配置成用于执行方法3100的帧生成逻辑和相关联的电路可以加载用于统一流预测网络(例如,如图30A-图30B中的统一神经网络3005)的外推权重,以将流预测网络配置成用于经由帧外推(3101)实现帧生成。然后,帧生成逻辑可以接收(3102)与渲染帧对相关联的RGB数据,连同那些帧之间的密集1:1光流。光流是由观察者和场景的相对运动引起的场景内的对象、表面和边缘的表观运动模式。光流可以是稀疏的或密集的。接收到的密集1:1光流指定输入帧之间的像素级运动。对于外推,输入帧对是给定的三个帧的集合中的帧一和帧二,并且帧生成逻辑将预测未来的帧三。例如,可以使用预测的未来的帧代替云游戏平台上的丢帧或代替延迟渲染帧以维持一致的帧速率。
帧生成逻辑和相关联的电路可以通过计算两个输入帧的总体均值并从每个帧减去总体均值来预处理(3104)输入帧。经预处理的帧和输入光流被串接(3106)并且被发送作为至流估计网络的输入。在一个实施例中,通过将输入帧的红色通道、绿色通道和蓝色通道连同2D光流的两个通道分层来执行输入帧和光流的串接,以创建八个通道的输入数据。该八个通道输入被提供至流预测网络。对于外推,流估计网络预测(3107)输入的帧二与未来的帧三之间的光流。然后,流预测网络输出输入的帧二与预测的帧三之间的预测的流量(Flow_23(流_23))。然后,帧生成逻辑可以使用预测的Flow_23来扭曲(3108)原始的输入帧二以预测帧三。然后,帧生成逻辑和相关联的电路可以输出(3109)预测的帧3。
如图31B中所示,被配置成用于执行方法3110的帧生成逻辑和相关联的电路可以加载用于统一流预测网络(例如,如图30A-图30B中的统一神经网络3005)的内插权重,以将流预测网络配置成用于经由帧内插(3111)实现帧生成。类似于在方法3100中,帧生成逻辑然后可以接收与渲染帧对相关联的RGB数据,连同那些帧之间的密集1:1光流(3112)。对于内插,输入帧对是给定的三个帧的集合中的帧一和帧三,并且帧生成逻辑将对中间的帧二进行内插,从而增加了该帧集合的有效帧速率。然后,帧生成逻辑和相关联的电路可以通过计算两个输入帧的总体均值并从每个帧减去总体均值来预处理(3114)输入帧。经预处理的帧和输入光流被串接(3116)并且被发送作为至流估计网络的输入。流估计网络预测(3117)输入的帧一与内插的帧二之间的光流。然后,流预测网络输出输入的帧二与内插的帧二之间的预测流(Flow_12)。然后,帧生成逻辑可以使用预测的Flow_12来扭曲(3118)原始输入的帧一以对经渲染的帧一与帧三之间的中间的帧二进行内插。然后,帧生成逻辑和相关联的电路可以输出(3119)经内插的帧二。
图32图示根据实施例的、用于帧外推和帧内插的光流预测网络3200。光流预测网络3200是图30A-图30B中的统一神经网络3005的一个实现方式的示例,其用于一个实施例中。其他实施例被配置成用于使用其他神经网络实现方式。光流预测网络3200对串接的输入3202进行操作,串接的输入3202可以如关于图31A-图31B的方法3100、3110所描述的那样被生成。在一个实施例中,串接的输入包括256×256×8的输入数据块(例如,8个通道的256×256数据),但是其他块大小在其他实施例中可以被使用。256×256×8的串接的输入包括两个输入帧中的每个输入帧的红色通道、绿色通道和蓝色通道,以及与两个输入帧之间的二维密集光流相关联的两个通道。光流预测网络3200可以以分片方式进行操作,以处理渲染帧的整个RGB集合和光流数据,从而预测用于要被生成的帧的经过滤的最终光流3240,其中多个片并发地被计算。串接的输入3202附加地包括针对网络指定通道维度d和滤波器大小k的参数。
在一个实施例中,光流预测网络3200具有像U-Net的体系结构,其具有下采样阶段3210A-3210J和上采样阶段3220A-3220F。下采样阶段3210A-3210J各自包括至少一个卷积层,以在增加特征通道的数量的同时降低输入数据的空间分辨率。使用越来越多的特征通道以越来越小的空间分辨率以两个跨度(s=2)或一个跨度(s=1)来执行卷积操作。上采样阶段3220A-3220F包括至少一个上卷积或转置卷积层,以减少特征通道的数量,同时将空间分辨率增加回到原始输入分辨率。由与上采样阶段3220A-3220F相关联的卷积层3221A-3221F生成一系列中间光流估计(流5-流0)。中间光流估计以多个分辨率被生成,并且使网络能够更好地维持RGB和光流数据之间的对准。
在操作期间,串接的输入3202在第一下采样阶段3210A之前被提供至编码器。相同的串接的输入3202在最终上采样阶段3220F之后、恰好在去噪滤波器之前被输入至解码器。附加地,来自所选择的下采样阶段(例如,下采样阶段3210A-3210C、3210E)的输出与相关联的上采样阶段的输出以及处于该分辨率的中间光流估计进行串接。该串接的输入用作用于下一上采样阶段的输入,并且这些步骤被重复直到经下采样的数据被上采样回到原始输入分辨率。处于输入分辨率的经上采样的输出与原始输入连同处于输入分辨率的流估计进行串接,以生成串接的中间输出。最终卷积阶段3224被应用于串接的中间输出以获得最终流估计。掩码计算阶段3222生成去噪滤波器掩码,其是根据处于输入分辨率的串接的中间输出来计算的。在去噪阶段3230处将去噪滤波器掩码应用于最终流估计,以生成经滤波的最终光流3240。该最终流估计可用于扭曲先前渲染帧,以取决于输入帧的顺序和被加载以用于在操作期间使用的权重的集合来生成内插帧或外推帧。
在训练期间,可以例如使用双线性内插、双三次内插或另一上采样算法将多个中间光流上采样到输入分辨率。然后,使用经上采样的中间光流来扭曲原始输入帧,以基于输入的帧一和帧二来生成未来的帧三的多个估计,或者基于输入的帧一和帧三来生成中间的帧二。对未来的帧或中间的帧的这些多重估计连同最终预测可以经由具有不同权重的损失函数与基准真值渲染帧进行比较,其中较低的权重被赋予根据要求更多上采样的光流构建的估计。该损失函数用于在训练期间执行中间监督,而不是仅依赖于基准真值与最终输出之间的比较,一旦网络完全被训练就会产生更高质量的推断结果。中间监督对总体网络具有规范化作用。
图33图示根据实施例的统一内插/外推网络的去噪部分3300的更详细视图。在一个实施例中,去噪部分3300是图32的光流预测网络3200的部分,并且对串接的中间输出3302进行操作(诸如是从最终卷积阶段3224输出的最终光流估计)。串接的中间输出3302是由串接的RGB和光流数据的高度、宽度和通道深度定义的3D张量,其在一个实施例中是(h×w×c=256×256×(8+0.25d+2))。掩码计算阶段3222通过应用第一卷积和ReLU阶段3312、第二卷积和ReLU阶段3314以及softmax激活阶段3316、基于串接的中间输出3302来生成去噪滤波器掩码。第二卷积和ReLU阶段3314输出多个k×k滤波器,每个像素一个k×k滤波器。softmax激活阶段3316对多个k×k个滤波器中的每个滤波器的权重进行归一化以生成去噪滤波器。然后,去噪阶段3230应用生成的去噪滤波器来生成经滤波的最终光流3240,该经滤波的最终光流3240是密集的1:1光流,其使得能够对渲染帧中的每个像素进行扭曲以生成新的外推帧或内插帧。
softmax激活是由以下等式(2)指定的函数:
图34A-图34B图示根据实施例的、用于训练和操作统一内插/外推网络的方法。图34A图示用于训练和部署统一内插/外推网络的方法3400。图34B图示操作统一内插/外推网络的方法3410。
如图34A中所示,方法3400包括执行操作以使用外推数据集针对统一外推/内插神经网络训练光流预测从而生成用于帧生成的外推权重(3402),并且还包括执行操作以使用内插数据集针对统一外推/内插神经网络训练光流预测从而生成用于帧生成的内插权重(3404)。然后,外推权重和内插权重可以连同统一外推/内插神经网络一起被部署(3406)。网络和相关联的权重可以由ISV连同使用帧生成的软件应用一起被部署。网络和相关联的权重还可以连同驱动器和实用套件一起由硬件供应商部署。在大型和多样化的数据集上被训练的通用权重集可以被使用,或者基于特定应用数据训练的特定应用权重可以被使用。
如图34B中所示,方法3410包括确定用于启用了帧生成的应用的帧生成模式(3412)。基于应用的类型和使用模型,应用可以被配置成用于特定的帧生成模式。应用还可以被配置有对两种帧生成模式的支持,其中特定模式可以基于针对应用的配置设置来选择。基于由应用开发者配置的或由应用的用户选择的帧生成模式,应用可以将帧生成模式配置成用于外推或内插(3413)。
当被配置成用于外推时,应用可以将预训练的外推权重加载(3414)到用于统一内插/外推网络的光流预测网络中。然后,统一内插/外推网络可以预测(3416)渲染帧与未来帧之间的光流。然后,预测的光流可用于扭曲(3418)渲染帧以生成未来帧。当被配置成用于内插时,应用可以将预训练的内插权重加载(3415)到用于统一内插/外推网络的光流预测网络中。然后,统一内插/外推网络可以预测(3417)用于两个渲染帧之间的中间帧的光流。然后,预测的光流可用于扭曲(3419)渲染帧以生成两个渲染帧之间的中间帧。对测试数据集执行的实验示出,从内插和外推两者得到的图像具有出色的峰值信噪比(PSNR)、组织类似性指数测量(SSIM)和均方根误差(RMSE)度量,其距基准真值图像仅有微小的偏差。
具有低分辨率光流的硬件高效的神经帧预测
上述技术使用1:1密集光流,针对每个1x1像素块具有光流向量。使用较低分辨率的光流(例如,2×2到8×8)可能会导致质量降级,尤其是在高频区域中。通过2×2,我们意指OF每个2px×2px(2像素×2像素)片进行平均,并且8×8意指OF每个8px×8px(8像素×8像素)片进行平均。这相应地降低了生成的OF的总体空间分辨率,对于2×2OF是4X,并且对于8×8OF是64X。
本文中描述的是使用较低分辨率光流(2×2、4×4、8×8)的网络训练方法,较低分辨率光流在使用RGB图像进行训练期间与1×1光流(伪基准真值)结合使用以生成具有渲染内容的未来帧。使用较低分辨率光流和全分辨率帧执行推断。经更新的网络以所需存储器和吞吐量的一小部分产生与1×1光流相当的结果。
这些技术特别适用于原生8×8运动向量且不具有密集光流加速硬件的硬件。这些技术也适用于其中期望高分辨率输出(4K/8K)而无需使用全分辨率光流以节省推断成本的场景。
图35A-图35B图示用于使用低分辨率光流实现推断的训练操作。图35A示出了用于训练神经网络模型3507以使用分辨率高于帧的RGB像素数据的分辨率的光流数据来执行帧生成的训练系统3500。神经网络模型3507可以是例如图30A-图30B的统一神经网络3005。神经网络模型3507的实现方式可以包括图32的光流预测网络3200和图33的相关联的去噪部分3300、以及连同光流预测网络3200一起训练的光流上采样和去噪网络。图35B示出了包括推断部署的系统3520,推断部署被配置成用于使用分辨率低于帧的RGB像素数据的分辨率的光流数据来执行帧生成。
如图35A中所示,在训练期间,提供8×8光流3502、连同1×1伪基准真值光流3504和RGB帧数据3506作为输入。1×1伪基准真值光流3504可以被增强或被诸如全分辨率运动向量之类的合适代理代替以引导帧预测。RGB帧数据3506可以被其他帧数据替换或增强,其他帧数据诸如法线图、深度缓冲器、反照率和运动向量,或者在帧的渲染期间被使用并且还用于训练帧生成的其他G缓冲器数据。8×8光流3502可以用针对每个n×n像素块(例如,4×4、2×2等)包括运动或光流向量的另一分辨率的光流来代替或补充。
在一个实施例中,8×8光流3502数据由例如图27的OFG2729生成。在一个实施例中,用于训练的1×1伪基准真值光流3504通过使用具有低于实时性能的高质量离线上采样算法对8×8光流3502训练数据进行上采样来生成。在一个实施例中,使用并非所有GPU模型上都存在的光流生成或光流上采样硬件来生成用于训练的1×1伪基准真值光流3504。训练神经网络模型3507以使用小于输入数据的源分辨率的光流数据实现帧生成在无法在运行时生成帧之间的密集光流数据的图形处理器上实现帧生成。来自神经网络的输出将是处于全分辨率3512的预测的光流,连同处于全分辨率3516的预测帧。图35A的训练训练神经网络模型3507内的光流上采样和去噪网络,神经网络模型3507使得8×8光流能够用于推断。
如图35B中所示,已经使用包括8×8光流3502和1×1伪基准真值光流3504的训练数据训练的训练神经网络3527可以基于包括运行时生成的8×8光流3522的输入、使用所包括的光流上采样和去噪网络来生成全分辨率预测的光流。在各实施例中,运行时生成的8×8光流3522可以用针对每个n×n像素块(例如,4×4、2×2等)包括运动或光流向量的另一分辨率的光流来代替或补充。在一个实施例中,运行时生成的n×n光流的分辨率与在训练经训练的神经网络3527时在训练数据集中使用的光流的最低分辨率相关。
图36图示根据实施例的、包括光流上采样的帧生成的方法3600。被配置成用于执行方法3600的帧生成逻辑和相关联的电路可以加载用于统一流预测网络(例如,如图30A-图30B中的统一神经网络3005)的外推权重,以将流预测网络配置成用于经由帧内插或外推实现帧生成(3601)。然后,帧生成逻辑可以接收(3602)与渲染帧对相关联的RGB数据、连同那些帧之间的子源分辨率光流(诸如例如,8×8光流)。帧生成逻辑可以使用流上采样和去噪网络将较低分辨率光流上采样到(3625)经去噪的全分辨率光流。流上采样和去噪网络输出对源分辨率为1:1的1×1光流,使得运动向量针对渲染帧的每个像素都存在。
帧生成逻辑和相关联的电路可以通过计算两个输入帧的总体均值并从每个帧减去总体均值来预处理(3604)输入帧。经预处理的帧和输入光流被串接(3606)并且被发送作为至流估计网络的输入。在一个实施例中,通过将输入帧的红色通道、绿色通道和蓝色通道连同2D光流的两个通道分层来执行输入帧和光流的串接,以创建八个通道的输入数据。该八个通道输入被提供至流预测网络。然后,帧生成逻辑和相关联的电路可以使用流预测网络、基于串接的RGB帧和光流来预测(3607)光流。对于外推,流估计网络预测输入的帧二与未来的帧三之间的光流。对于内插,流估计网络预测输入的帧一与内插的帧二之间的光流。然后,流预测网络输出从输入的帧二到预测的帧三的预测的光流(flow_23)或从输入的帧一到预测的帧二的预测的光流(flow_12)。然后,帧生成逻辑可以使用预测的光流来扭曲(3608)原始输入帧以对预测帧进行内插或外推。然后,帧生成逻辑和相关联的电路可以输出(3609)预测帧。
图37图示根据实施例的光流上采样和去噪网络3700。流上采样和去噪网络3700将滤波器内核大小k、和作为参数的上采样因子f、连同分辨率小于源分辨率的子源分辨率光流作为输入(3702)。可以被使用的示例性内核大小为k=3,使得光流上采样和去噪网络3700将预测3×3滤波器。子源分辨率光流包括用于渲染帧的每n×n个像素的运动向量,诸如当n=8时用于每8×8个像素的运动向量。
在一个实施例中,网络由两个阶段组成:滤波器预测器阶段3710和滤波器应用阶段3720。在滤波器预测器阶段3710中,对于滤波器大小k,网络针对经上采样的光流的每个像素预测k×k滤波器内核(3712)。例如,输入光流的形式可以是高度(h)×宽度(w)×2。滤波器预测器阶段3710可以针对上采样因子f以h*f×w*f×2的形式预测与经上采样的光流的每个像素相对应的k×k滤波器内核。使用softmax对滤波器内核进行归一化,使得权重总和为1(3714)。在滤波器应用阶段3720中,网络首先用输入的上采样因子对输入的较低分辨率光流进行上采样,以根据上采样因子f生成中间全分辨率光流(3704)。使用局部预测的k×k内核对中间全分辨率光流的每个像素进行滤波(3706)。滤波器应用阶段3720的输出是最终经上采样且经去噪的光流3708。最终经上采样且经去噪的光流3708可用作至光流预测网络3200的输入,如图32中所示。
在一个实施例中,流上采样和去噪网络3700与图32的光流预测网络3200中的在图33中描绘的去噪部分3300不同地被配置。流上采样和去噪网络3700根据输入预测由局部3×3(例如,对于k=3)滤波器组成的滤波器掩码。在光流被上采样到源分辨率之后,流上采样和去噪网络3700使用预测的滤波器对光流进行去噪。由此,上采样和去噪网络3700使得能够对小于帧数据的源分辨率的输入光流进行上采样以及去噪。然后,将经上采样且经去噪的光流数据用作至光流预测网络的输入,以出于帧生成的目的来执行光流估计。相反,光流预测网络3200的去噪部分3300根据计算最终光流估计以供在帧生成中使用的阶段之前的阶段的输出来预测局部滤波器。然后,将最终光流估计去噪为用于经由内插或外推实现帧生成的最终光流估计。
图38A-图38B图示根据实施例的、训练和操作具有光流上采样和去噪子网络以及光流估计子网络的神经网络的方法。图38A图示训练神经网络以结合光流预测使用内插或外推来执行光流上采样和去噪的方法3800。图38B图示根据实施例的、使用低分辨率光流的帧生成的方法3810。
如图38A中所示,方法3800包括用于使用外推数据集执行对神经帧预测网络的端到端训练从而生成用于帧生成和超分辨率的外推权重的操作(3802),并且还包括用于使用内插数据集执行对神经帧预测网络的端到端训练从而生成用于帧生成和超分辨率的内插权重的操作(3804)。然后,外推权重和内插权重可以连同神经帧预测网络一起被部署(3806)。训练逻辑可以被配置成用于利用帧数据和光流数据集来训练用于神经帧预测网络的光流预测,光流数据包括基于图像损失和光流损失的n×n和1x1光流。在训练期间,n×n光流被上采样到1×1光流,并且然后与用作伪基准真值的1×1密集光流训练数据进行比较。此比较用于生成“流”损失。预测帧还与基准真值帧进行比较,以生成“图像”损失。将流损失和图像损失相加以生成总损失。然后,对总损失进行反向传播以更新网络的参数。在一个实施例中,诸如VGG损失或感知损失之类的损失被用于改善预测帧的质量。
网络和相关联的权重可以由ISV连同使用帧生成的软件应用一起被部署。网络和相关联的权重还可以连同驱动器和实用套件一起由硬件供应商部署。在一个实施例中,部署包括向ISV提供模型的实现方式和经训练的权重的集合以包括在利用帧生成的3D应用中。在一个实施例中,部署在默认实现方式和权重与图形驱动程序版本一起被提供的情况下包括模型的实现方式以及软件开发工具包(software development kit,SDK)或驱动程序版本中的经训练的权重的集合。在广义数据集上被训练的通用权重集可以被使用,或者基于特定应用数据训练的特定应用权重可以被使用。
如图38B中所示,方法3810可以由图形处理器内的帧生成逻辑和相关联的电路执行,以使帧生成能够使用以比用于帧的像素数据的分辨率更低的分辨率生成的光流来执行。方法3810包括,在神经网络模型的部署实例处,接收(3812)包括多个帧的帧数据和多个帧中的帧之间的n×n光流的输入。多个帧中的帧之间的n×n光流可以是例如2×2、4×4或8×8光流,其包括用于每个2×2、4×4或8×8像素块的运动向量。输入光流数据可以包括从渲染引擎输出的运动向量和/或由图形处理器内的光流加速器或视频编码器生成的运动向量或光流数据以及其他G缓冲区数据。光流上采样去噪网络(诸如图37的光流上采样和去噪网络3700)可以对作为至1×1光流的输入而被接收的n×n光流进行上采样,该1×1光流的分辨率与渲染帧数据的分辨率相同(3814)。
然后,神经网络模型的部署实例可以使用经上采样的光流来预测(3816)用于生成帧的光流。取决于作为输入被提供的帧数据和光流以及被加载到网络中的权重,预测可以是内插或外推。然后,神经网络模型的部署实例可以使用预测的光流来扭曲(3818)渲染帧以生成预测帧。例如,可以预测光流,该光流使得三帧序列中的帧一能够基于帧一、帧三、以及帧一与帧三之间的光流被扭曲以对该序列中的帧二进行内插。可以预测光流,该光流使得三帧序列中的帧二能够基于帧一、帧二、以及帧一与帧二之间的光流被扭曲以对该序列中的帧三进行外推。
附加的示例性计算设备
图39是根据实施例的包括图形处理器3904的计算设备3900的框图。计算设备3900的多个版本可以是通信设备或可被包括在通信设备内,该通信设备诸如机顶盒(例如,基于互联网的有线电视机顶盒等)、基于全球定位系统(global positioning system,GPS)的设备等。计算设备3900还可以是移动计算设备或可被包括在移动计算设备内,该移动计算设备诸如,蜂窝电话、智能电话、个人数字助理(personal digital assistant,PDA)、平板计算机、膝上型电脑、电子阅读器、智能电视、电视平台、可穿戴设备(例如,眼镜、手表、项链、智能卡、首饰、服饰等)、媒体播放器等。例如,在一个实施例中,计算设备3900包括采用集成电路(integrated circuit,“IC”)(诸如,片上系统(“SoC”或“SOC”))的移动计算设备,该集成电路将计算设备3900的各种硬件和/或软件部件集成在单个芯片上。
计算设备3900包括图形处理器3904。图形处理器3904表示本文中描述的任何图形处理器。在一个实施例中,图形处理器3904包括缓存3914,该缓存3914可以是单个缓存,或可被划分为缓存存储器的多个片段,该缓存3614包括但不限于任何数量的L1缓存、L2缓存、L3缓存或L4缓存、渲染缓存、深度缓存、采样器缓存、和/或着色器单元缓存。在一个实施例中,缓存3914可以是与应用处理器3906共享的最后一级缓存。在一个实施例中,计算设备3900包括CXL逻辑3912,以促进应用处理器3906与图形处理器3904之间的数据共享和传输。计算设备3900还可以包括硬件和软件逻辑3913,以实现可缩放I/O虚拟化(scalable I/Ovirtualization,S-IOV)或单根I/O虚拟化(single-root I/O virtualization,SRIOV),以向由应用处理器3906执行的软件域提供图形处理器3904的虚拟实例。
在一个实施例中,图形处理器3904包括图形微控制器,该图形微控制器实现用于图形处理器的控制和调度逻辑。控制和调度逻辑可以是由图形微控制器3915执行的固件。固件可以在引导时由图形驱动器逻辑3922加载。固件也可以被编程到电子可擦除可编程只读存储器,或从图形微控制器3915内的闪存存储器设备加载。固件可以启用GPU OS 3916,该GPU OS 3916包括设备管理逻辑3917和驱动器逻辑3918、以及调度器3919。GPU OS 3916还可包括图形存储器管理器3920,该图形存储器管理器3920可补充或替代图形驱动器逻辑3922内的图形存储器管理器3921。
图形处理器3904还包括GPGPU引擎3944,该GPGPU引擎3944包括如本文中所描述的一个或多个图形引擎、图形处理器核心、以及其他图形执行资源。此类图形执行资源能以包括但不限于以下各项的形式来呈现:执行单元、着色器引擎、片段处理器、顶点处理器、图形多处理器、流式多处理器、图形处理器集群、或适于处理图形资源或图像资源或在异构处理器(包括集成或分立的图形或和/或并行处理元件)中执行通用计算操作的计算资源的任何集合。GPGPU引擎3944的处理资源可被包括在连接到衬底的硬件逻辑的多个片内。GPGPU引擎3944可包括GPU片3945,这些GPU片3945包括图形处理和执行资源、缓存、采样器等。GPU片3945还可以包括本地易失性存储器或者可以与一个或多个存储器片耦合。在一个实施例中,GPU片3945还可以包括媒体引擎,该媒体引擎包括可配置成用于基于两个输入帧生成至少稀疏光流的CODEC(编解码器)硬件。
GPGPU引擎3944还可包括一个或多个特殊片3946,这一个或多个特殊片3946包括例如非易失性存储器片3956、网络处理器片3957、和/或通用计算片3958。GPGPU引擎3944还包括矩阵乘法加速器3960。通用计算片3958还可包括用于加速矩阵乘法操作的逻辑。非易失性存储器片3956可包括非易失性存储器单元和控制器逻辑。非易失性存储器片3956的控制器逻辑可由设备管理逻辑3917或驱动器逻辑3918中的一个来管理。网络处理器片3957可包括网络处理资源,该网络处理资源耦合至计算设备3900的输入/输出(I/O)源3910内的物理接口。网络处理器片3957可由设备管理逻辑3917或驱动器逻辑3918中的一个或多个来管理。
在一个实施例中,矩阵乘法加速器3960是模块化可缩放稀疏矩阵乘法加速器。矩阵乘法加速器3960可包括多个处理路径,其中每个处理路径包括多个管线阶段。每个处理路径可执行单独的指令。在各实施例中,矩阵乘法加速器3960可具有本文中描述的矩阵乘法加速器中的任何一个或多个的体系结构特征。例如,在一个实施例中,矩阵乘法加速器3960是脉动阵列,该脉动阵列能配置成用于利用四的倍数个逻辑阶段(例如,四个、八个、十二个、十六个等)进行操作。在一个实施例中,矩阵乘法加速器3960包括具有四级管线的两路径矩阵乘法加速器或具有两级管线的四路径矩阵乘法加速器中的一个或多个实例。在一个实施例中,矩阵乘法加速器3960包括被配置为可缩放稀疏矩阵乘法加速器的处理元件。矩阵乘法加速器3960可被用于加速经由XMX扩展或促进矩阵计算操作的加速的另一计算库而被执行的矩阵操作。在该示例中,矩阵乘法加速器3960可用于执行与本文中描述的光流估计、帧生成和神经超分辨率网络的卷积和上卷积/转置卷积层相关联的操作,其可被包括在由计算设备3900使用的统一神经网络模型3923中。
如所图示,在一个实施例中,除了图形处理器3904之外,计算设备3900可进一步包括任何数量和类型的硬件部件和/或软件部件,包括但不限于应用处理器3906、存储器3908、以及输入/输出(I/O)源3910。应用处理器3906可与硬件图形管线交互以共享图形管线功能。经处理的数据被存储在硬件图形管线中的缓冲器中,并且状态信息被存储在存储器3908中。所得的数据可以被传递至显示控制器以用于经由显示设备来输出。显示设备可具有各种类型,诸如,阴极射线管(Cathode Ray Tube,CRT)、薄膜晶体管(Thin FilmTransistor,TFT)、液晶显示器(Liquid Crystal Display,LCD)、有机发光二极管(OrganicLight Emitting Diode,OLED)阵列等,并且显示设备可被配置成用于经由图形用户界面来向用户显示信息。
应用处理器3906可包括诸如图1的(一个或多个)处理器102之类的一个或多个处理器,并且可以是至少部分地用于执行计算设备3900的操作系统(OS)3902的中央处理单元(CPU)。OS 3902可充当计算设备3900的硬件和/或物理资源与一个或多个用户之间的接口。OS 3902可包括用于计算设备3900中的各种硬件设备的驱动器逻辑。驱动器逻辑可包括图形驱动器逻辑3922,该图形驱动器逻辑3922可包括用户模式图形驱动器和/或内核模式图形驱动器。图形驱动器逻辑可包括图形存储器管理器3921,用于管理用于图形处理器3904的虚拟存储器地址空间。图形存储器管理器3921可促进可由应用处理器3906和图形处理器3904访问的统一虚拟地址空间。
构想了在一些实施例中,图形处理器3904可作为应用处理器3906的部分(诸如,作为物理CPU封装的部分)而存在,在这种情况下,存储器3908的至少部分可由应用处理器3906和图形处理器3904共享,但是存储器3908的至少部分对于图形处理器3904可以是独占的,或者图形处理器3904可具有存储器的单独存储。存储器3908还可以经由CXL逻辑3912与分立版本的图形处理器3904共享。
存储器3908可包括缓冲器的预分配区域(例如,帧缓冲器);然而,应当由本领域普通技术人员理解的是,实施例不限于此,并且可使用可由较低的图形管线访问的任何存储器。存储器3908可包括包含利用图形处理器3904以渲染桌面或3D图形场景的应用的各种形式的随机访问存储器(random-access memory,RAM)(例如,SDRAM、SRAM等)。存储器控制器中枢可访问存储器3908中的数据,并且将其转发至图形处理器3904以供图形管线处理。存储器3908可变得对计算设备3900内的其他部件可用。例如,在软件程序或应用的实现方式中,从计算设备3900的各种I/O源3910接收到的任何数据(例如,输入图形数据)在由一个或多个处理器(例如,应用处理器3906)操作之前可临时被排队到存储器3908中。类似地,软件程序确定应当从计算设备3900通过计算系统接口中的一个计算系统接口被发送到外部实体或者应当被存储到内部存储元件内的数据在被传送或被存储之前通常临时被排队在存储器3908中。
I/O源可包括诸如触摸屏、触摸面板、触摸板、虚拟或常规键盘、虚拟或常规鼠标、端口、连接器、网络设备等之类的设备,并且可经由平台控制器中枢进行附接。附加地,I/O源3910可包括被实现用于将数据传递到计算设备3900和/或从计算设备3900传递数据的一个或多个I/O设备(例如,网络适配器);或者被实现用于计算设备3900内的大规模非易失性存储装置(例如,SSD/HDD)的一个或多个I/O设备。包括字母数字和其他键的用户输入设备可用于将信息和命令选择传递到图形处理器3904。另一类型的用户输入设备是用于将方向信息和命令选择传递到GPU并用于控制显示设备上的光标移动的光标控件,诸如,鼠标、轨迹球、触摸屏、触摸板、或光标方向键。可采用计算设备3900的相机和话筒阵列来观察姿势、记录音频和视频,并且接收和传送可视命令和音频命令。
I/O源3910可包括一个或多个网络接口。网络接口可包括相关联的网络处理逻辑和/或与网络处理器片3957耦合。一个或多个网络接口可提供对以下各项的访问:LAN、广域网(wide area network,WAN)、城域网(metropolitan area network,MAN)、个域网(personal area network,PAN)、蓝牙、云网络、蜂窝或移动网络(例如,第三代(3rdGeneration,3G)、第四代(4th Generation,4G)、第五代(5th Generation,5G)等)、内联网、互联网等。(一个或多个)网络接口可包括例如具有一个或多个天线的无线网络接口。(一个或多个)网络接口也可包括例如用于经由网络电缆与远程设备通信的有线网络接口,网络电缆可以是例如以太网电缆、同轴电缆、光纤电缆、串行电缆或并行电缆。
(一个或多个)网络接口可例如通过遵照IEEE 802.11标准来提供对LAN的访问,并且/或者无线网络接口可例如通过遵照蓝牙标准来提供对个域网的访问。还可支持其他无线网络接口和/或协议,包括先前版本和后续版本的标准。附加于或替代于经由无线LAN标准的通信,(一个或多个)网络接口可使用例如以下协议来提供无线通信:时分多址(TimeDivision,Multiple Access,TDMA)协议、全球移动通信系统(Global Systems for MobileCommunications,GSM)协议、码分多址(Code Division,Multiple Access,CDMA)协议和/或任何其他类型的无线通信协议。
应当领会的是,对于某些实现方式,比在上文描述的示例中更少或更多地配备的系统可以是优选的。因此,取决于多种因素,本文中描述的计算设备的配置可因实现方式而异,这些因素诸如,价格约束、性能要求、技术改进、或其他情况。示例包括(但不限于)移动设备、个人数字助理、移动计算设备、智能电话、蜂窝电话、手机、单向寻呼机、双向寻呼机、消息收发设备、计算机、个人计算机(personal computer,PC)、桌面型电脑、膝上型电脑、笔记本计算机、手持式计算机、平板计算机、服务器、服务器阵列或服务器场、web服务器、网络服务器、互联网服务器、工作站、小型计算机、大型计算机、超级计算机、网络装置、web装置、分布式计算系统、多处理器系统、基于处理器的系统、消费电子产品、可编程消费电子产品、电视机、数字电视、机顶盒、无线接入点、基站、订户站、移动订户中心、无线电网络控制器、路由器、中枢、网关、桥接器、交换机、机器、或其组合。
实施例可以被提供为例如计算机程序产品,该计算机程序产品可包括一种或多种机器可读介质,该一种或多种机器可读介质具有被存储于其上的机器可执行指令,该机器可执行指令在由一个或多个机器(诸如,计算机、计算机的网络或其他电子设备)执行时可使得该一个或多个机器执行根据本文中所描述的实施例的操作。机器可读介质可包括但不限于:软盘、光盘、CD-ROM(致密盘只读存储器)以及磁光盘、ROM、RAM、EPROM(可擦除可编程只读存储器)、EEPROM(电可擦除可编程只读存储器)、磁卡或光卡、闪存存储器、或者适合于存储机器可执行指令的其他类型的介质/机器可读介质。
此外,实施例可作为计算机程序产品被下载,其中,经由通信链路(例如,调制解调器和/或网络连接)、借助于在载波或其他传播介质中实例化和/或由载波或其他传播介质调制的一个或多个数据信号,可将程序从远程计算机(例如,服务器)传输至作出请求的计算机(例如,客户端)。
贯穿本文档,术语“用户”可以可互换地被称为“观看者”、“观察者”、“人”、“个人”、“终端用户”,等等。应当注意,贯穿本文档,如“图形域”之类的术语可以与“图形处理单元”、“图形处理器”或简称为“GPU”可互换地引用,并且类似地,“CPU域”或“主机域”可以与“计算机处理单元”、“应用处理器”或简称为“CPU”可互换地引用。
应当注意的是,贯穿本文档,可以可互换地使用如“节点”、“计算节点”、“服务器”、“服务器设备”、“云计算机”、“云服务器”、“云服务器计算机”、“机器”、“主机”、“设备”、“计算设备”、“计算机”、“计算系统”等之类的术语。应当进一步注意,贯穿本文档,可以可互换地使用如“应用”、“软件应用”、“程序”、“软件程序”、“包”、“软件包”等之类的术语。并且,贯穿本文档,可以可互换地使用如“作业”、“输入”、“请求”、“消息”等之类的术语。
构想了贯穿本文档可以互换地引用如“请求”、“查询”、“作业”、“工作”、“工作项”和“工作负载”之类的术语。类似地,“应用”或“代理”可以指代或包括通过应用编程接口(application programming interface,API)提供的计算机程序、软件应用、游戏、工作站应用等,API诸如自由渲染API,诸如开放图形库(Open Graphics Library,)、开放计算语言(Open Computing Language,)、11、12等,其中“调遣”可以互换地称为“工作单元”或“绘制”,并且类似地,“应用”可以互换地称为“工作流”或简单地称为“代理”。例如,诸如三维(three-dimensiona,3D)游戏的工作负载之类的工作负载可以包括并发出任何数量和类型的“帧”,其中,每个帧可以表示图像(例如,帆船、人脸)。进一步地,每个帧可以包括并提供任何数量和类型的工作单元,其中,每个工作单元可以表示由其对应帧表示的图像(例如,帆船、人脸)的一部分(例如,帆船的桅杆、人脸的前额)。然而,为了一致性,贯穿本文档,每一项可以由单个术语(例如,“调遣”、“代理”等)来引用。
本文中对“一个实施例”、“实施例”、“示例实施例”等的引用指示所描述的实施例可包括特定的特征、结构或特性,但是每个实施例可以不一定包括该特定的特征、结构或特性。而且,此类短语不一定是指同一实施例。此外,当结合实施例描述特定的特征、结构或特性时,认为结合无论是否被明确描述的其他实施例而影响此类特征、结构或特性是在本领域技术人员的知识范围之内的。
在上文描述的各实施例中,除非另外专门指出,否则,诸如短语“A、B或C中的至少一个”之类的分隔语言旨在被理解为意味着A、B、或C、或其任何组合(例如,A、B、和/或C)。由此,分隔语言不旨在也不应当被理解为暗示给定的实施例要求A中的至少一个、B中的至少一个或C中的至少一个各自都存在。类似地,以“A、B或C中的至少一者”的形式列出的项可以意指(A);(B);(C);(A和B);(B和C);或(A、B和C)。
本文中提供的技术的某些方面包括可以以算法的形式或与算法相关地描述的逻辑和相关联的操作。应当注意的是,此类逻辑可以具体化在软件、固件和/或硬件中。当逻辑具体化在软件中时,此类逻辑可以被下载以驻留在由各种操作系统使用的不同平台上并从这些不同的平台操作,并且由处理器执行以执行相关联的操作。在固件中具体化的逻辑可以在本文中描述的微控制器或处理器设备上执行。当逻辑具体化在硬件中时,此类逻辑可以是数字逻辑的形式。此类数字逻辑还可以与模拟电路相关联。
在一些实施例中,如“显示屏”和“显示表面”之类的术语可互换地用于指代显示设备的可见部分,而显示设备的其余部分可以被嵌入到诸如智能电话、可穿戴设备等之类的计算设备中。构想到并且应当注意,实施例不限于任何特定的计算设备、软件应用、硬件部件、显示设备、显示屏或表面、协议、标准等。例如,实施例可被应用于任何数量和类型的计算机上的任何数量和类型的实时应用,并且可以与任何数量和类型的计算机上的任何数量和类型的实时应用一起使用,任何数量和类型的计算机诸如桌面型电脑、膝上型电脑、平板计算机、智能电话、头戴式显示器、以及其他可穿戴设备等。进一步地,例如,使用这种新颖技术渲染场景以实现高效性能的范围可以从诸如桌面合成之类简单场景到诸如3D游戏、增强现实应用等之类的复杂场景。
本文中描述了用于经由神经帧生成和神经超采样来增强针对3D渲染应用的用户体验的技术。一个实施例提供了用于帧内插和外推的等待时间感知统一神经网络。这种统一神经网络将内插和外推网络合并成一个广义网络,该广义网络可被应用于内插和外推两者,这取决于可接受的性能的等待时间。另一实施例提供硬件高效的且等待时间感知的时空神经帧预测。硬件高效且等待时间感知的时空神经帧预测使用单个网络实现帧生成和机器学习超采样两者,而不是将单独的网络用于帧生成和超采样。
一个实施例提供了一种数据处理系统,该数据处理系统包括:存储器设备,该存储器设备被配置成用于存储指令;以及并行处理器,该并行处理器包括电路,该电路被配置成用于执行矩阵操作。并行处理器被配置成用于:加载与机器学习模型相关联的外推权重,该机器学习模型能配置成用于经由外推和内插中的一个来估计光流;执行与机器学习模型相关联的操作,该操作用于基于外推权重、以及多个渲染帧、多个渲染帧之间的光流来估计预测的光流。然后,并行处理器可以基于多个渲染帧中的渲染帧和预测的光流来外推预测帧。并行处理器可以被配置成用于基于预测的光流来扭曲多个渲染帧中的渲染帧以外推预测帧。
在一个实施例中,并行处理器被配置成用于:加载与机器学习模型相关联的内插权重,机器学习模型能被配置成用于经由外推和内插中的一个来估计光流;执行与机器学习模型相关联的操作,操作用于基于内插权重、多个渲染帧、多个渲染帧之间的光流来估计内插光流;以及基于多个渲染帧中的渲染帧和内插光流,在多个预测帧中的帧之间内插中间帧。并行处理器可以被配置成用于:基于内插光流来扭曲多个渲染帧中的渲染帧以内插中间帧。
在一个实施例中,并行处理器被配置成用于:接收多个渲染帧的红色、绿色和蓝色(RGB)颜色数据以及多个渲染帧之间的光流;对多个渲染帧的RGB颜色数据进行预处理,以生成经预处理的RGB数据;将经预处理的RGB数据和多个渲染帧之间的光流串接成多通道输入数据块;以及经由与机器学习模型相关联的操作和加载的内插权重或外推权重、基于多通道输入数据块来估计预测的光流或内插光流。为了对RGB颜色数据进行预处理,并行处理器被配置成用于:计算多个渲染帧的总体均值;以及从多个渲染帧中的每个帧减去总体均值。并行处理器可以经由被配置成用于执行矩阵操作的电路来估计预测的光流和内插光流。为了估计预测的光流和内插光流中的一个,并行处理器被配置成用于:将多通道输入数据块加载到被配置成用于执行矩阵操作的电路中;以及经由与机器学习模型相关联的矩阵操作、根据通道维度参数和滤波器大小来处理多通道输入数据块。
为了处理多通道输入数据块,并行处理器被配置成用于:经由与机器学习模型的编码器部分相关联的操作、根据通道维度参数来处理多通道输入数据块。编码器部分包括第一多个卷积层。然后,并行处理器可以经由与机器学习模型的解码器部分相关联的操作、根据通道维度参数来处理编码器部分的输出,其中,解码器部分包括多个转置卷积层和第二多个卷积层。然后,并行处理器可以经由与机器学习模型的去噪部分相关联的操作、根据滤波器大小参数来处理解码器部分的输出,解码器部分用于输出预测的光流或内插光流。
在一个实施例中,为了处理编码器部分的输出,并行处理器被配置成用于:生成与多个转置卷积层中的第一转置卷积层相关联的中间输出;对中间输出执行与第二多个卷积层中的第一卷积层相关联的操作,以估计中间光流;以及将中间光流与经由解码器部分与编码器部分之间的跳跃连接接收到的中间输出和相关联的解码器阶段输出串接,以生成针对多个转置卷积层中的第二转置卷积层的输入。
一个实施例提供了一种方法,该方法包括:使用外推数据集执行对时空神经帧预测网络的端到端训练从而生成用于帧生成和超分辨率的外推权重;使用内插数据集执行对时空神经帧预测网络的端到端训练从而生成用于帧生成和超分辨率的内插权重;以及将外推权重和内插权重连同时空神经帧预测网络一起部署,时空神经帧预测网络用于使图形处理器能够以源分辨率执行内插帧生成或外推帧生成,并且将渲染帧和预测帧上采样到目标分辨率。
在一个实施例中,使用外推数据集来执行对时空神经帧预测网络的端到端训练包括:对与多个渲染帧中的第一帧和第二帧相关联的红色、绿色和蓝色(RGB)数据进行预处理,多个渲染帧处于源分辨率;对时空神经帧预测网络中的第一子网络执行前向传递,第一子网络用于基于第一帧和第二帧生成外推第三帧,外推第三帧处于源分辨率;对时空神经帧预测网络中的第二子网络执行前向传递,第二子网络用于对外推第三帧进行上采样以生成经上采样的外推第三帧,经上采样的外推第三帧处于目标分辨率。
方法可附加地包括:基于处于源分辨率的基准真值第三帧、以及外推第三帧来生成源分辨率损失数据;基于处于目标分辨率的基准真值第三帧、以及经上采样的外推第三帧来生成目标分辨率损失数据;将源分辨率损失数据和目标分辨率损失数据相加,以生成总损失数据;以及反向传播总损失数据,以更新第二子网络和第一子网络。
在一个实施例中,使用内插数据集来执行对时空神经帧预测网络的端到端训练包括:对与多个渲染帧中的第一帧和第三帧相关联的红、绿和蓝(RGB)数据进行预处理,多个渲染帧处于源分辨率;对时空神经帧预测网络的第一子网络执行前向传递,第一子网络用于基于第一帧和第三帧生成内插第二帧,内插第二帧处于源分辨率;对时空神经帧预测网络的第二子网络执行前向传递,第二子网络用于对内插第二帧进行上采样以生成经上采样的内插第二帧,经上采样的内插第二帧处于目标分辨率。
在一个实施例中,方法附加地包括:基于处于源分辨率的基准真值第二帧、以及内插第二帧来生成源分辨率损失数据;基于处于目标分辨率的基准真值第二帧、以及内插第二帧来生成目标分辨率损失数据;将源分辨率损失数据和目标分辨率损失数据相加,以生成总损失数据;以及反向传播总损失数据,以更新第二子网络和第一子网络。
对第一子网络执行前向传递可以包括:生成处于小于源分辨率的中间光流估计;将中间光流估计上采样到源分辨率;经由中间光流估计来扭曲多个渲染帧中的渲染帧以生成中间扭曲帧;生成处于源分辨率的最终光流估计;经由最终光流扭曲多个渲染帧中的渲染帧以生成最终扭曲帧;以及至少部分地基于中间扭曲帧和最终扭曲帧与基准真值渲染帧的比较、经由损失函数来生成损失。
方法可附加地包括:生成处于多个不同分辨率的多个中间光流估计,每个分辨率都小于源分辨率;经由多个中间光流估计扭曲多个渲染帧中的渲染帧,以生成多个中间扭曲帧;以及至少部分地基于多个中间扭曲帧和最终扭曲帧与基准真值渲染帧的比较、经由加权损失函数来生成损失,加权损失函数基于不同的中间光流估计的各自的分辨率针对多个中间光流估计中的不同的中间光流估计具有不同的权重。
对第二子网络执行前向传递可以包括:生成处于小于目标分辨率的中间上采样帧;将中间上采样帧上采样到目标分辨率;生成处于目标分辨率的最终上采样帧;以及至少部分地基于中间上采样帧和最终上采样帧与处于目标分辨率的基准真值渲染帧的比较、经由损失函数来生成损失。
在一个实施例中,方法附加地包括:生成处于多个不同分辨率的多个中间上采样帧,每个分辨率都小于目标分辨率;将多个中间上采样帧中的每个中间上采样帧上采样到目标分辨率,以生成多个中间上采样帧;以及至少部分地基于多个中间上采样帧和最终上采样帧与基准真值渲染帧的比较、经由加权损失函数来生成损失,加权损失函数基于不同的中间上采样帧的各自的分辨率针对多个中间上采样帧中的不同的中间上采样帧具有不同的权重。
一个实施例提供了一种非暂态机器可读介质,该非暂态机器可读介质存储指令,该指令当由一个或多个处理器执行时,使得一个或多个处理器执行包括本文中所描述的方法中的任一方法的操作。此外,本文中描述的技术可以具体化在包括用于执行本文中所描述的任何方法的装置的系统中。
一个实施例提供了一种图形处理器,包括:系统接口;以及与系统接口耦合的多个处理资源,该多个处理资源被配置成用于:渲染多个帧的帧数据,帧数据在多个帧中的帧之间具有相关联的光流数据,帧数据以源分辨率渲染;对多个帧的帧数据内的颜色数据进行预处理,以生成经预处理的颜色数据;将经预处理的颜色数据和光流数据串接成多通道输入数据块;基于多通道输入数据块、经由与机器学习模型相关联的第一操作以及用于机器学习模型的内插权重或外推权重来估计多个帧中的帧与新帧之间的光流;基于多个帧中的帧与新帧之间的估计的光流,将多个帧中的帧的颜色数据扭曲到新帧中;以及经由与机器学习模型相关联的第二操作将多个帧的帧数据和新帧上采样到目标分辨率。
在一个实施例中,多个帧中的帧之间的光流数据的分辨率低于源分辨率。在此类实施例中,多个处理资源可以被配置成用于:将多个帧中的帧之间的光流数据上采样到源分辨率;以及将经上采样的光流数据串接成多通道输入数据块。
本文中附加地描述了用于经由神经帧生成和光流数据的神经上采样针对3D渲染应用增强用户体验的技术。在一个实施例中,使用稀疏光流数据和密集光流数据两者来训练神经网络以使得神经帧生成能够由部署的神经网络仅使用稀疏光流数据来执行。稀疏光流数据可以由经训练的神经网络上采样到密集光流数据。神经网络可以使用经上采样的密集光流数据来执行帧生成。
一个实施例提供了一种数据处理系统,包括:存储器设备,该存储器设备被配置成用于存储指令;并行处理器,该并行处理器包括电路,该电路被配置成用于执行矩阵操作,该并行处理器被配置成用于:加载与机器学习模型相关联的外推权重或内插权重,该机器学习模型能配置成用于经由外推和内插中的一个来估计光流;执行与机器学习模型相关联的操作,操作用于基于加载的权重、处于第一分辨率的多个渲染帧、以及处于第一分辨率的多个渲染帧之间的光流来估计预测的光流,光流从处于第二分辨率的输入光流被上采样到第一分辨率,第二分辨率小于第一分辨率;以及基于多个渲染帧中的渲染帧和预测的光流、经由外推和内插中的一个来生成预测帧。
在一个实施例中,并行处理器被配置成用于:基于预测的光流来扭曲多个渲染帧中的渲染帧以生成预测帧。在一个实施例中,并行处理器被配置成用于:经由与机器学习模型的光流上采样和去噪子网络相关联的操作,将输入光流从第二分辨率上采样到第一分辨率;以及经由由针对光流上采样和去噪子网络执行的操作预测的去噪滤波器对经上采样的光流进行去噪。在一个实施例中,并行处理器被配置成用于:基于处于第二分辨率的输入光流来预测去噪滤波器。
在一个实施例中,并行处理器被配置成用于:接收多个渲染帧的红色、绿色和蓝色(RGB)颜色数据;接收多个渲染帧之间的光流,光流包括经上采样且经去噪的光流;对多个渲染帧的RGB颜色数据进行预处理,以生成经预处理的RGB数据;将经预处理的RGB数据和多个渲染帧之间的光流串接成多通道输入数据块;以及经由与机器学习模型相关联的操作和加载的权重、基于多通道输入数据块来生成预测的光流。在一个实施例中,为了对RGB颜色数据进行预处理,并行处理器被配置成用于:计算多个渲染帧的总体均值;以及从多个渲染帧中的每个帧减去总体均值。
在一个实施例中,并行处理器被配置成用于:对输入光流进行上采样和去噪,并且经由被配置成用于执行矩阵操作的电路生成预测的光流。在一个实施例中,为了对输入光流进行上采样和去噪,并行处理器被配置成用于:将包括处于第二分辨率的输入光流的光流数据加载到被配置成用于执行矩阵操作的电路中;基于光流数据预测去噪滤波器;根据上采样因子对光流数据进行上采样,以生成处于第一分辨率的中间光流数据;以及将预测的去噪滤波器应用于中间光流数据,以生成处于第一分辨率的经上采样且经去噪的光流数据。在一个实施例中,为了生成预测的光流,并行处理器被配置成用于:将多通道输入数据块加载到被配置成用于执行矩阵操作的电路中;以及经由与机器学习模型相关联的矩阵操作、根据通道维度参数和滤波器大小来处理多通道输入数据块。
在一个实施例中,为了处理多通道输入数据块,并行处理器被配置成用于:经由与机器学习模型的编码器部分相关联的操作、根据通道维度参数来处理多通道输入数据块,编码器部分包括第一多个卷积层;经由与机器学习模型的解码器部分相关联的操作、根据通道维度参数来处理编码器部分的输出,解码器部分包括多个转置卷积层和第二多个卷积层;以及经由与机器学习模型的去噪部分相关联的操作、根据滤波器大小参数来处理解码器部分的输出,解码器部分用于输出预测的光流或内插光流。
在一个实施例中,为了处理编码器部分的输出,并行处理器被配置成用于:生成与多个转置卷积层中的第一转置卷积层相关联的中间输出;对中间输出执行与第二多个卷积层中的第一卷积层相关联的操作,以估计中间光流;以及将中间光流与经由解码器部分与编码器部分之间的跳跃连接接收到的中间输出和相关联的解码器阶段输出串接,以生成针对多个转置卷积层中的第二转置卷积层的输入。
一个实施例提供了一种方法,该方法包括:使用外推数据集执行对神经帧预测网络的端到端训练,以生成用于帧生成和光流上采样的外推权重;使用内插数据集执行对神经帧预测网络的端到端训练,以生成用于帧生成和光流上采样的内插权重;以及将外推权重和内插权重连同神经帧预测网络一起部署,神经帧预测网络用于使图形处理器能够执行包括以下各项的操作:将光流数据从处于第二分辨率的输入光流上采样到第一分辨率,第二分辨率小于第一分辨率;对被上采样到第一分辨率的光流数据进行去噪;以及经由处于第一分辨率的经上采样的光流数据、经由内插和外插中的一个来生成帧。
在一个实施例中,使用外推数据集或内插数据集来执行对神经帧预测网络的端到端训练包括:对与多个渲染帧相关联的红色、绿色和蓝色(RGB)数据进行预处理,多个渲染帧处于第一分辨率;对神经帧预测网络中的第一子网络执行前向传递,第一子网络用于将光流数据从第二分辨率上采样到第一分辨率并且用于对经上采样的光流数据进行去噪;以及对神经帧预测网络中的第二子网络执行前向传递,第二子网络用于基于处于第一分辨率的多个渲染帧以及由第一子网络生成的经上采样且经去噪的光流数据来预测渲染帧与生成帧之间的光流。
该方法的一个实施例附加地包括:基于处于第一分辨率的基准真值光流、以及由第一子网络生成的经上采样且经去噪的光流数据来生成光流损失数据;基于处于第一分辨率的基准真值渲染帧、以及由第二子网络经由内插和外推中的一个生成的帧来生成图像损失数据;将光流损失数据和图像损失数据相加,以生成总损失数据;以及反向传播总损失数据,以更新第二子网络和第一子网络。
在一个实施例中,对第二网络执行前向传递包括:生成小于第一分辨率的中间光流估计;将中间光流估计上采样到第一分辨率;经由中间光流估计来扭曲多个渲染帧中的渲染帧,以生成中间扭曲帧;生成处于第一分辨率的最终光流估计;经由最终光流估计来扭曲多个渲染帧中的渲染帧,以生成最终扭曲帧;以及至少部分地基于中间扭曲帧和最终扭曲帧与基准真值渲染帧的比较、经由损失函数来生成损失。
一个实施例提供了一种非暂态机器可读介质,该非暂态机器可读介质存储指令,该指令当由一个或多个处理器执行时,使得一个或多个处理器执行本文中所描述的方法的操作。一个实施例提供了一种系统,该系统包括用于执行本文中所描述的方法的装置。
一个实施例提供了一种图形处理器,包括:系统接口;以及与系统接口耦合的多个处理资源,该多个处理资源被配置成用于:渲染多个帧的帧数据,帧数据在多个帧中的帧之间具有相关联的光流数据,帧数据处于第一分辨率并且光流数据处于第二分辨率,第二分辨率小于第一分辨率;将处于第二分辨率的光流数据上采样到处于第一分辨率的光流数据;经由基于处于第二分辨率的光流数据生成的滤波器对处于第一分辨率的光流数据进行去噪;经由与机器学习模型相关联的第一操作以及用于机器学习模型的内插权重和外推权重中的一个来估计多个帧中的帧与新帧之间的光流,光流经由处于第一分辨率的经上采样且经去噪的光流数据来估计;以及基于多个帧中的帧与新帧之间的估计的光流,将多个帧中的帧的颜色数据扭曲到新帧中。
在一个实施例中,多个处理资源被配置成用于:基于包括处于第二分辨率的光流数据的输入来预测去噪滤波器;以及将预测的去噪滤波器应用于被上采样到第一分辨率的光流数据,以生成处于第一分辨率的经上采样且经去噪的光流数据。在一个实施例中,多个处理资源被配置成用于:对多个帧的帧数据内的颜色数据进行预处理,以生成经预处理的颜色数据;以及将经预处理的颜色数据和经上采样且经去噪的光流数据串接成多通道输入数据块。在一个实施例中,为了对多个帧的帧数据内的颜色数据进行预处理,多个处理资源被配置成用于:计算多个渲染帧的总体均值颜色;以及从多个渲染帧中的每个帧中减去总体均值颜色。在一个实施例中,多个处理资源被配置成用于:基于多通道输入数据块来估计多个帧中的帧与新帧之间的光流。
前述说明书和附图应以说明性意义而非限制性意义来看待。本领域技术人员将理解,可对本文中描述的实施例作出各种修改和改变,而不背离如所附权利要求所述的特征的更宽泛的精神和范围。

Claims (22)

1.一种数据处理系统,包括:
存储器设备,所述存储器设备被配置成用于存储指令;
并行处理器,所述并行处理器包括电路,所述电路被配置成用于执行矩阵操作,所述并行处理器被配置成用于:
加载与机器学习模型相关联的外推权重或内插权重,所述机器学习模型能配置成用于经由外推和内插中的一个来估计光流;
执行与所述机器学习模型相关联的操作,所述操作用于基于加载的权重、处于第一分辨率的多个渲染帧、以及处于所述第一分辨率的所述多个渲染帧之间的光流来估计预测的光流,光流从处于第二分辨率的输入光流被上采样到所述第一分辨率,所述第二分辨率小于所述第一分辨率;以及
基于所述多个渲染帧中的渲染帧和所述预测的光流、经由外推和内插中的一个来生成预测帧。
2.如权利要求1所述的数据处理系统,所述并行处理器被配置成用于:基于所述预测的光流来扭曲所述多个渲染帧中的渲染帧以生成预测帧。
3.如权利要求2所述的数据处理系统,所述并行处理器被配置成用于:
经由与所述机器学习模型的光流上采样和去噪子网络相关联的操作,将所述输入光流从所述第二分辨率上采样到所述第一分辨率;以及
经由由针对所述光流上采样和去噪子网络执行的操作预测的去噪滤波器对经上采样的光流进行去噪。
4.如权利要求3所述的数据处理系统,所述并行处理器被配置成用于:基于处于所述第二分辨率的所述输入光流来预测所述去噪滤波器。
5.如权利要求3或4所述的数据处理系统,所述并行处理器被配置成用于:
接收所述多个渲染帧的红色、绿色和蓝色(RGB)颜色数据;
接收所述多个渲染帧之间的光流,光流包括经上采样且经去噪的光流;
对所述多个渲染帧的RGB颜色数据进行预处理,以生成经预处理的RGB数据;
将所述经预处理的RGB数据和所述多个渲染帧之间的光流串接成多通道输入数据块;以及
经由与所述机器学习模型相关联的操作和所述加载的权重、基于所述多通道输入数据块来生成所述预测的光流。
6.如权利要求5所述的数据处理系统,其中,为了对所述RGB颜色数据进行预处理,所述并行处理器被配置成用于:
计算所述多个渲染帧的总体均值;以及
从所述多个渲染帧中的每个帧减去所述总体均值。
7.如权利要求5所述的数据处理系统,所述并行处理器被配置成用于对所述输入光流进行上采样和去噪,并且经由被配置成用于执行矩阵操作的所述电路生成所述预测的光流。
8.如权利要求7所述的数据处理系统,其中,为了对所述输入光流进行上采样和去噪,所述并行处理器被配置成用于:
将包括处于所述第二分辨率的所述输入光流的光流数据加载到被配置成用于执行矩阵操作的所述电路中;
基于所述光流数据来预测去噪滤波器;
根据上采样因子对所述光流数据进行上采样,以生成处于所述第一分辨率的中间光流数据;以及
将预测的去噪滤波器应用于所述中间光流数据,以生成处于所述第一分辨率的经上采样且经去噪的光流数据。
9.如权利要求7所述的数据处理系统,其中,为了生成所述预测的光流,所述并行处理器被配置成用于:
将所述多通道输入数据块加载到被配置成用于执行矩阵操作的电路中;以及
经由与所述机器学习模型相关联的矩阵操作、根据通道维度参数和滤波器大小来处理所述多通道输入数据块。
10.如权利要求9所述的数据处理系统,其中,为了处理所述多通道输入数据块,所述并行处理器被配置成用于:
经由与所述机器学习模型的编码器部分相关联的操作、根据所述通道维度参数来处理所述多通道输入数据块,所述编码器部分包括第一多个卷积层;
经由与所述机器学习模型的解码器部分相关联的操作、根据所述通道维度参数来处理所述编码器部分的输出,所述解码器部分包括多个转置卷积层和第二多个卷积层;以及
经由与所述机器学习模型的去噪部分相关联的操作、根据所述滤波器大小参数来处理所述解码器部分的输出,所述解码器部分用于输出所述预测的光流或内插光流。
11.如权利要求10所述的数据处理系统,其中,为了处理所述解码器部分的输出,所述并行处理器被配置成用于:
生成与所述多个转置卷积层中的第一转置卷积层相关联的中间输出;
对所述中间输出执行与所述第二多个卷积层中的第一卷积层相关联的操作,以估计中间光流;以及
将所述中间光流与经由所述解码器部分与所述编码器部分之间的跳跃连接接收到的所述中间输出和相关联的解码器阶段输出串接,以生成针对所述多个转置卷积层中的第二转置卷积层的输入。
12.一种方法,包括:
使用外推数据集执行对神经帧预测网络的端到端训练,以生成用于帧生成和光流上采样的外推权重;
使用内插数据集执行对所述神经帧预测网络的端到端训练,以生成用于帧生成和光流上采样的内插权重;以及
将外推权重和内插权重连同所述神经帧预测网络一起部署,所述神经帧预测网络用于使图形处理器能够执行包括以下各项的操作:
将光流数据从处于第二分辨率的输入光流上采样到第一分辨率,所述第二分辨率小于所述第一分辨率;
对被上采样到所述第一分辨率的光流数据进行去噪;以及
经由处于所述第一分辨率的经上采样的光流数据、经由内插和外推中的一个来生成帧。
13.如权利要求12所述的方法,其中,使用所述外推数据集或所述内插数据集来执行对所述神经帧预测网络的端到端训练包括:
对与多个渲染帧相关联的红色、绿色和蓝色(RGB)数据进行预处理,所述多个渲染帧处于所述第一分辨率;
对所述神经帧预测网络中的第一子网络执行前向传递,所述第一子网络用于将所述光流数据从所述第二分辨率上采样到所述第一分辨率并且用于对经上采样的光流数据进行去噪;以及
对所述神经帧预测网络的第二子网络执行前向传递,所述第二子网络用于基于处于所述第一分辨率的所述多个渲染帧以及由所述第一子网络生成的经上采样且经去噪的光流数据来预测渲染帧与生成帧之间的光流。
14.如权利要求13所述的方法,进一步包括:
基于处于所述第一分辨率的基准真值光流、以及由所述第一子网络生成的所述经上采样且经去噪的光流数据来生成光流损失数据;
基于处于所述第一分辨率的基准真值渲染帧、以及由所述第二子网络经由内插和外推中的一个生成的帧来生成图像损失数据;
将所述光流损失数据和所述图像损失数据相加,以生成总损失数据;以及
反向传播所述总损失数据,以更新所述第二子网络和所述第一子网络。
15.如权利要求14所述的方法,其中,对所述第二子网络执行所述前向传递包括:
生成小于所述第一分辨率的中间光流估计;
将所述中间光流估计上采样到所述第一分辨率;
经由所述中间光流估计来扭曲所述多个渲染帧中的渲染帧,以生成中间扭曲帧;
生成处于所述第一分辨率的最终光流估计;
经由所述最终光流估计来扭曲所述多个渲染帧中的渲染帧,以生成最终扭曲帧;以及
至少部分地基于所述中间扭曲帧和所述最终扭曲帧与基准真值渲染帧的比较、经由损失函数来生成损失。
16.一种非暂态机器可读介质,所述非暂态机器可读介质存储指令,所述指令当由一个或多个处理器执行时,使得所述一个或多个处理器用于执行包括如权利要求12-15中的任一项所述的方法的操作。
17.一种系统,包括用于执行如权利要求12-15中的任一项所述的方法的装置。
18.一种图形处理器,包括:
系统接口;以及
与所述系统接口耦合的多个处理资源,所述多个处理资源被配置成用于:
渲染多个帧的帧数据,所述帧数据在所述多个帧中的帧之间具有相关联的光流数据,所述帧数据处于第一分辨率并且所述光流数据处于第二分辨率,所述第二分辨率小于所述第一分辨率;
将处于所述第二分辨率的光流数据上采样到处于所述第一分辨率的光流数据;
经由基于处于所述第二分辨率的光流数据生成的滤波器对处于所述第一分辨率的光流数据进行去噪;
经由与机器学习模型相关联的第一操作以及用于所述机器学习模型的内插权重和外推权重中的一个来估计所述多个帧中的帧与新帧之间的光流,所述光流经由处于所述第一分辨率的经上采样且经去噪的光流数据来估计;以及
基于所述多个帧中的帧与所述新帧之间的估计的光流,将所述多个帧中的帧的颜色数据扭曲到所述新帧中。
19.如权利要求18所述的图形处理器,所述多个处理资源被配置成用于:
基于包括处于所述第二分辨率的光流数据的输入来预测去噪滤波器;以及
将预测的去噪滤波器应用于被上采样到第一分辨率的光流数据,以生成处于所述第一分辨率的经上采样且经去噪的光流数据。
20.如权利要求19所述的图形处理器,所述多个处理资源被配置成用于:
对所述多个帧的帧数据内的颜色数据进行预处理,以生成经预处理的颜色数据;以及
将经预处理的颜色数据和所述经上采样且经去噪的光流数据串接成多通道输入数据块。
21.如权利要求20所述的图形处理器,其中,为了对所述多个帧的帧数据内的颜色数据进行预处理,所述多个处理资源被配置成用于:
计算所述多个渲染帧的总体均值颜色;以及
从所述多个渲染帧中的每个帧减去所述总体均值颜色。
22.如权利要求20所述的图形处理器,所述多个处理资源被配置成用于:基于所述多通道输入数据块来估计所述多个帧中的帧与所述新帧之间的光流。
CN202311697886.5A 2023-03-16 2023-12-11 使用低分辨率光流的硬件高效神经帧预测 Pending CN118674842A (zh)

Applications Claiming Priority (3)

Application Number Priority Date Filing Date Title
US63/490,609 2023-03-16
US18/308,725 US20240311962A1 (en) 2023-03-16 2023-04-28 Hardware-efficient neural frame prediction with low resolution optical flow
US18/308,725 2023-04-28

Publications (1)

Publication Number Publication Date
CN118674842A true CN118674842A (zh) 2024-09-20

Family

ID=92720260

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202311697886.5A Pending CN118674842A (zh) 2023-03-16 2023-12-11 使用低分辨率光流的硬件高效神经帧预测

Country Status (1)

Country Link
CN (1) CN118674842A (zh)

Similar Documents

Publication Publication Date Title
CN112905240B (zh) 用于脉动阵列上的块稀疏操作的架构
CN117546200A (zh) 在图形环境中具有重要性采样的组合的降噪和升尺度网络
EP4152162B1 (en) Immediate offset of load store and atomic instructions
CN117529751A (zh) 使用内核飞溅网络的时域摊销的超采样
WO2022271227A1 (en) Dual pipeline parallel systolic array
EP4432227A1 (en) Hardware-efficient neural frame prediction with low resolution optical flow
CN116091680A (zh) 用于超采样的输入滤波和采样器加速
CN116091332A (zh) 经由程序性着色性输出扩充运动向量
WO2022271230A1 (en) Systolic array of arbitrary physical and logical depth
WO2023022806A1 (en) Temporally amortized supersampling using a mixed precision convolutional neural network
EP4428823A1 (en) Unified latency aware neural network for frame interpolation and prediction
EP4432241A1 (en) Optical flow mip adjustment for rendering and encoding
EP4478197A1 (en) Coarse and fine filtering for gpu hardware-based performance monitoring
CN118674603A (zh) 经由时间感知机器学习模型的基于时间的帧生成
CN117581259A (zh) 图形数据的联合去噪和超采样
CN117859148A (zh) 使用混合精度卷积神经网络的时域摊销的超采样
US12367549B2 (en) Motion vector refinement for temporally amortized supersampling
EP4432239A1 (en) Preserving g-buffer and optical flow in uv space
US20240312028A1 (en) Optical flow generation using rasterized triangle id buffers
US20240312113A1 (en) Uv space rendering and ai processing
EP4498237A1 (en) 32-bit channel-aligned integer multiplication via multiple multipliers per-channel
EP4432240A1 (en) Methodology to enable highly responsive gameplay in cloud and client gaming
EP4553649A1 (en) Multiple register allocation sizes for gpu hardware threads
EP4177824A1 (en) Motion vector refinement for temporally amortized supersampling
EP4498236A1 (en) Avoiding the use of a result crossbar when down converting to packed register formats

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication