Skip to content

Feature Request: CPU performance optimization with MKL-DNN #4186

@mingfeima

Description

@mingfeima

Hi, our team works on DL frameworks performance optimization on CPU.
PyTorch CPU performance can be significantly improved with MKL-DNN, an open source deep neural network library on Intel CPU platform. Currently MKL-DNN supports Conv2d, Relu, BatchNorm, Pooling, Concat, etc. The CPU performance improvement will benefit cloud users as well as HPC researchers who have access to Xeon clusters.
Will PyTorch be interested in the part?

cc @gujinghui @PenghuiCheng @XiaobingSuper @jianyuh @VitalyFedyunin

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: mkldnnRelated to Intel IDEEP or oneDNN (a.k.a. mkldnn) integrationtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions