Move oneDNN Graph python bindings

intel / ideep

Intel® Optimization for Chainer*, a Chainer module providing numpy like API and DNN acceleration using MKL-DNN.

MIT License

165 stars 90 forks source link

Closed sanchitintel closed 6 months ago

sanchitintel commented 8 months ago

Python bindings of oneDNN Graph should be in the include directory.

sanchitintel commented 8 months ago

Hi @Xia-Weiwen, can you please review this PR, and include it in your PyTorch oneDNN v3.4 upgrade PR? Thanks!

Xia-Weiwen commented 8 months ago

Hi @sanchitintel. Have you tested the change with PyTorch? Are there any potential impacts? Is it needed for IPEX?

Also cc @guobing-chen @jgong5 @chunyuan-w @yanbing-j for review. Thanks.

sanchitintel commented 8 months ago

Hi @Xia-Weiwen, there's no impact, since this header is currently not being used.

It'd be used for using oneDNN Graph python bindings with stock PyTorch Inductor in the near future.