Skip to content

[Tracker] XNNPACK BF16 Support #19854

@GregoryComer

Description

@GregoryComer

This is the top-level tracking task for support for the BF16 dtype in the XNNPACK delegate. Most of the work is upstream in XNNPACK itself, but we will need to do a small amount of wiring up in the ET delegate once the upstream changes are complete.

Task List:

  • Support BF16 for GEMMs, element-wise ops, data movement, and reduction ops in upstream XNNPACK.
  • Update the ET XNNPACK dependency to pick up the BF16 changes.
  • Wire up BF16 support in the ET XNNPACK partitioner and serialization format.
  • Wire up BF16 support in the ET XNNPACK runtime.

Metadata

Metadata

Assignees

Labels

No labels
No labels
No fields configured for Feature.

Projects

Status

In progress

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions