[NNC] enable bf16 for mkldnn prepack conv2d

[chunyuan-w]

Pitch

Enable bf16 support for mkldnn prepack conv2d in NNC.

Performance

The BF16 conv performance has been evaluated in https://github.com/pytorch/pytorch/pull/82705.

Additional context

This PR depends on BF16 support in NNC: https://github.com/pytorch/pytorch/pull/84041.

[pytorch-bot[bot]]

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/85394

• :page_facing_up: Preview Python docs built from this PR
• :page_facing_up: Preview C++ docs built from this PR
• :question: Need help or want to give feedback on the CI? Visit our office hours

Note: Links to docs will display an error until the docs builds have been completed.

:white_check_mark: No Failures, 1 Pending

As of commit 093001d65b0d4e935937feed382c991496bfe3fe: :green_heart: Looks good so far! There are no failures yet. :green_heart:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[facebook-github-bot]

/easycla

As part of the transition to the PyTorch Foundation, this project now requires contributions be covered under the new CLA. See #85559 for additional details.

This comment will trigger a new check of this PR. If you are already covered, you will simply see a new "EasyCLA" check that passes. If you are not covered, a bot will leave a new comment with a link to sign.

[linux-foundation-easycla[bot]]

• :x: - login: @chunyuan-w / name: Chunyuan WU . The commit (1157878f51389a2510cbb2937d44bb1556ad8efc, 59de1d99819ced92e3b7ffe09082bd6022aac4e7, 1df4b1405076b69b3933545fb3e8e9b48aa20a10, 322b204a50cfed3eaf406e1bb63a534deb74bb39, 093001d65b0d4e935937feed382c991496bfe3fe) is not authorized under a signed CLA. Please click here to be authorized. For further assistance with EasyCLA, please submit a support request ticket.