You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2022/02/17 01:23:21 UTC

[GitHub] [tvm] yangulei commented on pull request #10112: [TIR, Relay] improve bfloat16 support

yangulei commented on pull request #10112:
URL: https://github.com/apache/tvm/pull/10112#issuecomment-1042474807


   Hi @billishyahao ,
   You can checkout [my fork](https://github.com/yangulei/tvm/tree/dev_byoc). The [readme file](https://github.com/yangulei/tvm/blob/dev_byoc/bench_oneDNN_BYOC/README.md) include a brief introduction about how to setup the environment and do the benchmarking. Be aware that this is not the final branch for upstreaming. You need a CPU with AVX-512 enabled to support bfloat16 functionally, and a CPU with AMX enabled to support bfloat16 natively, take a look at [oneDNN document](https://oneapi-src.github.io/oneDNN/dev_guide_data_types.html) for more details.
   FYI: we plan to upstream a tutorial about BYOC-oneDNN after the PRs about this are merged.


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org