XPU 到底是什么?

人工智能 术语 火炬 执行 资源请求 硬件
2021-10-25 02:46:40

我知道 CPU、GPU 和 TPU。但是,这是我第一次从PyTorch 关于 MODULE 的文档中了解 XPU 。

xpu(device=None)


Moves all model parameters and buffers to the XPU.

This also makes associated parameters and buffers different objects. So it should be called before constructing optimizer if the module will live on XPU while being optimized.

Note: This method modifies the module in-place.

Parameters

    device (int, optional) – if specified, all parameters will be copied to that device

Returns

    self

Return type

    Module

CPU代表中央处理器。GPU 代表图形处理单元。TPU 代表张量处理单元。

我们大多数人都知道,这些处理单元在人工智能的一些计算密集型领域(包括深度学习)的研究中非常有用。所以,我想知道它是否XPU在人工智能研究中也有用,因为它在 PyTorch 中使用。

从上下文来看,我可以说它PU代表processing unit. 但我不知道是什么X

XPU 的完整形式是什么?我在哪里可以详细了解 XPU?

1个回答

XPU是英特尔异构计算架构的设备抽象,可以映射到CPU、GPU、FPGA等加速器。XPU 中的“X”就像数学中的变量一样,因此您可以执行 X=C 并获得 CPU 加速,或者 X=G 获得 GPU 加速……这就是这个抽象名称背后的直觉。

为了集成一个新的加速器,你需要两件事: