[BYOC] Enable bfloat16 in DNNL BYOC - #11111

Merged

masahi merged 18 commits into

apache:mainfrom

yangulei:upstream_byoc_bf16

May 26, 2022

Commits on May 26, 2022

refine the code style (#10112)
yangulei
committed
686a9ff
support more data types in oneDNN BYOC
yangulei
committed
3b2b8e2
consider dtype when query layout
yangulei
committed
1fe4c4d
support more translation of blocked layout
yangulei
committed
bfcfe62
refine log for invalid layout transform
yangulei
committed
bc3d8ca
reset N and C for the weights
yangulei
committed
82c49c3
support multi-blocking in TransDims2Plain()
yangulei
committed
b5be62e
add tests for bf16 oneDNN BYOC
yangulei
committed
6ff6613
unregister 'round' OP in oneDNN BYOC
yangulei
committed
f7832c9
restore the criteria for fp32 tests
yangulei
committed
52fac7e
disable test_prune_dnnl_subgraph for bf16
yangulei
committed
6e194dc
fix typo in dnnl.py
yangulei
committed
43f6e78
delete tag::format_tag_last
yangulei
committed
388341d
delete 'is_weight' in layout2tag()
yangulei
committed
e91c23e
reuse dtype_dl2dnnl()
yangulei
committed
a161d2e
fix lint errors
yangulei
committed
a9d0ad8
change to WARNING for invalid laytout transform
yangulei
committed
a42764a
skip bf16 tests if AVX512 is unavailable
yangulei
committed
4caede4