Add bfloat16 to scalar_type.fbs#16882
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/16882
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New Failure, 10 PendingAs of commit 99ca1a7 with merge base b729d35 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
There was a problem hiding this comment.
Pull request overview
This pull request activates the BFLOAT16 scalar type in the FlatBuffer schema files by uncomment it from the enum definition. The change enables deserialization of programs that use bfloat16 tensors.
Changes:
- Activated BFLOAT16 enum value (15) in three scalar_type.fbs files
- Removed commented-out BFLOAT16 entries from the "not implemented" section
Reviewed changes
Copilot reviewed 3 out of 3 changed files in this pull request and generated 1 comment.
| File | Description |
|---|---|
| schema/scalar_type.fbs | Added BFLOAT16 = 15 enum entry and removed commented version |
| extension/flat_tensor/serialize/scalar_type.fbs | Added BFLOAT16 = 15 enum entry and removed commented version |
| devtools/etdump/scalar_type.fbs | Added BFLOAT16 = 15 enum entry and removed commented version |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
bef6493 to
99ca1a7
Compare
Summary
This should cover all the scalar_type.fbs files: https://github.com/search?q=repo%3Apytorch%2Fexecutorch%20path%3Ascalar_type.fbs&type=code
Test plan
Deserialize a program with bfloat16. See: https://www.internalfb.com/manifold/explorer/executorch/tree/models/whisper/test/model.pte