Update: convert all paged_attention examples from float16 to bfloat16#490
Open
hw-native-sys-bot wants to merge 1 commit intohw-native-sys:mainfrom
Open
Update: convert all paged_attention examples from float16 to bfloat16#490hw-native-sys-bot wants to merge 1 commit intohw-native-sys:mainfrom
hw-native-sys-bot wants to merge 1 commit intohw-native-sys:mainfrom
Conversation
|
Warning You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again! |
hw-native-sys-bot
force-pushed
the
sync/a5-pa-dtype-bfloat16
branch
from
17ba6df to
b7547dd
Compare
hw-native-sys-bot
changed the title
hw-native-sys-bot
force-pushed
the
sync/a5-pa-dtype-bfloat16
branch
from
b7547dd to
2d76a78
Compare
Convert all four paged_attention example variants to use bfloat16: - a2a3/tensormap_and_ringbuffer/paged_attention - a5/tensormap_and_ringbuffer/paged_attention - a2a3/host_build_graph/paged_attention - a5/host_build_graph/paged_attention Changes across all variants: - Change half/float16 to bfloat16_t/bfloat16 in kernel files - Update golden.py dtype from "float16" to "bfloat16" - Rename f16/fp16 variables and comments to bf16 - Align pto::Stride to Stride (matching reference style)
hw-native-sys-bot
force-pushed
the
sync/a5-pa-dtype-bfloat16
branch
from
2d76a78 to
44d638e
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
a2a3/tensormap_and_ringbuffer/paged_attentiona5/tensormap_and_ringbuffer/paged_attentiona2a3/host_build_graph/paged_attentiona5/host_build_graph/paged_attentionhalf/float16types tobfloat16_t/bfloat16in all kernel files"float16"to"bfloat16"f16/fp16variables and comments tobf16pto::StridetoStride(matching reference style)Testing