search

intel / ideep

Intel® Optimization for Chainer*, a Chainer module providing numpy like API and DNN acceleration using MKL-DNN.
MIT License
163 stars 90 forks source link

question: mkl-dnn upgrade for PyTorch #41

Open kimoni opened 4 years ago

kimoni commented 4 years ago

Hi Intel ideep team, thanks for the awesome works of integrating mkl-dnn with PyTorch.😃

I noticed that the pytorch_dnnl branch is used for such integration, while the latest commit shows the ideep leverages the 1.5 version of mkl-dnn. 

$ cd /root/pytorch/third_party/ideep
$ git log
commit 938cc68897bb46b8d4b228966edd9e23e471cf3b (HEAD, origin/pytorch_dnnl)
Author: pinzhenx <pinzhen.xu@intel.com>
Date:   Tue Jun 16 18:54:21 2020 +0000

    bump onednn to v1.5

Our application runs PyTorch on Intel CPU architecture, and we relies on mkl-dnn to gain better performances.

We want to know when will ideep upgrades mkl-dnn for PyToch, do you have a timetable for doing so?

CaoZhongZ commented 4 years ago

We update oneDNN according to its release, after verifying and notifying PyTorch team.