Skip to content

Updated the ROPE dtype for custom_dtype#989

Merged
quic-hemagnih merged 7 commits into
quic:mainfrom
asmigosw:bf16_rotary_dtype
May 19, 2026
Merged

Updated the ROPE dtype for custom_dtype#989
quic-hemagnih merged 7 commits into
quic:mainfrom
asmigosw:bf16_rotary_dtype

Conversation

@asmigosw
Copy link
Copy Markdown
Contributor

ROPE dtype was set to torch.get_default_dtype() which was float32 by default. Changed it to set the config's dtype which was set by user.

asmigosw added 4 commits May 15, 2026 13:50
Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
@asmigosw asmigosw requested review from quic-hemagnih and quic-rishinr and removed request for quic-rishinr May 18, 2026 06:48
Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
@asmigosw asmigosw requested a review from quic-mamta May 18, 2026 07:08
Copy link
Copy Markdown
Contributor

@quic-hemagnih quic-hemagnih left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@asmigosw I think the medium severity comments posted by Qgenie are valid and are edge/negative case handling which is a good coding practise.
Can you please incorporate both of them in your code.

asmigosw added 2 commits May 18, 2026 14:35
Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
Copy link
Copy Markdown
Contributor

@quic-hemagnih quic-hemagnih left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, lets merge once CI is passed

@quic-hemagnih quic-hemagnih merged commit af6865a into quic:main May 19, 2026
5 of 6 checks passed
quic-mohmeh pushed a commit to quic-mohmeh/efficient-transformers that referenced this pull request May 25, 2026
ROPE dtype was set to torch.get_default_dtype() which was float32 by
default. Changed it to set the config's dtype which was set by user.

---------

Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
Signed-off-by: Mohit Mehta <mohmeh@qti.qualcomm.com>
smedhe pushed a commit to smedhe/QEff_Sharvari that referenced this pull request May 25, 2026
ROPE dtype was set to torch.get_default_dtype() which was float32 by
default. Changed it to set the config's dtype which was set by user.

---------

Signed-off-by: Asmita Goswami <asmigosw@qti.qualcomm.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants