DeepSpeed
Fix the half-precision version of rotary_pos_emb kernel
#1683
Merged

Fix the half-precision version of rotary_pos_emb kernel #1683

RezaYazdaniAminabadi
fix the half-precision version of rotary_pos_emb kernel
4de49c34
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from awan-10 awan-10 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from cli99 cli99 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from conglongli conglongli 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from eltonzheng eltonzheng 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from jeffra jeffra 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from minjiaz minjiaz 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from samyam samyam 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from ShadenSmith ShadenSmith 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi requested a review from tjruwase tjruwase 4 years ago
RezaYazdaniAminabadi RezaYazdaniAminabadi enabled auto-merge (squash) 4 years ago
jeffra
jeffra approved these changes on 2022-01-10
RezaYazdaniAminabadi RezaYazdaniAminabadi merged 762e697a into master 4 years ago
mrwyattii mrwyattii deleted the gptj-inference/fix-half-kernels branch 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone