IPEX 的性能

Performance with IPEX

我在 PyTorch 中尝试了线性回归和梯度下降。 (https://www.analyticsvidhya.com/blog/2021/08/linear-regression-and-gradient-descent-in-pytorch/)

我已经使用以下命令导入了 Ipex:

import intel_pytorch_extension as ipex

如何将模型转换为ipex?我如何检查性能是否有所提高?有什么帮助吗?

导入IPEX后,定义模型时使用以下命令:

def model(X, IPEX=True):
return X @ w.t() + b

我们已经验证了您分享的 link 中的代码。您可以将 model(X) 替换为 model(X, IPEX=True)

IPEX 的性能有了显着提高。如果没有 IPEX,我们可以看到实际目标值和模型预测之间存在一些差异。其中,与 IPEX 值没有区别。