Add AXPY oneDNN handler

[arogowie-intel]

PR types

Performance optimization

PR changes

OPs

Describe

This PR introduces oneDNN AXPY handler providing BLAS::AXPY interface. The use of this handler is enabled at this moment only for bfloat16 data type for sum operator with SelectedRows input tensors. It gives almost 10% performance improvement on CPX machine.

Additionally this PR speeds up SGD operator with Selected rows with AXPY handler.

[paddle-bot-old]

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[paddle-bot-old]

Sorry to inform you that c7b8f49's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually.

[arogowie-intel]

CI-APPROVAL failure is because of need for approval to skip test:

2021-06-10 21:52:47 0. Unittest is not allowed to be disabled.
2021-06-10 21:52:47 You must have one RD (kolinwei(Recommend), wanghuancoder or luotao1) approval for the usage of @unittest.skip or @unittest.skipIf.
2021-06-10 21:52:47 +@**unittest.skipIf(not core.supports_bfloat16()**

[jczaja]

LGTM

[arlesniak]

LGTM

[arlesniak]

LGTM

[arogowie-intel]

@lidanqing-intel PR-CI-APPROVAL failed because of use of unittest.skipIf(not core.supports_bfloat16() and I need sb's approval for this.
@jczaja PR-CI-OP-benchmark fails in GPU tests for reasons not related to this PR. I rerun it for the third time.

[lidanqing-vv]

LGTM

[lidanqing-vv]

mkldnn bf16 ops tests always use unittest.skipIf(not core.supports_bfloat16()
Just like in other bf16 ops PRs https://github.com/PaddlePaddle/Paddle/pull/33172/files
so it can be approved.

[lidanqing-vv]

Hi, @arogowie-intel I cherry-picked your PR and it's been merged. You may close this.
By inheritance, you may avoid unittest.skipIf(not core.supports_bfloat16(). Since now Luotao left, approval requires other Baidu people to review and take long time.

[lidanqing-vv]

#33632 has merged this PR. So I close this one because we need to summarise merged PRs each week.