Commit Graph

2 Commits (fd398fae1a1cfce62afaf17d131cb994e35604df)

Author SHA1 Message Date
ytxiong fd398fae1a
refactor(rotaryEmbedding): refactor forward (#120)
* use fp16 in instruction (#80)

* delete torch_dtype of README's example code (#100)

* refactor the forward for rotary embedding

---------

Co-authored-by: WRH <12756472+wangruohui@users.noreply.github.com>
Co-authored-by: x54-729 <45304952+x54-729@users.noreply.github.com>
2023-07-25 15:25:48 +08:00
Sun Peng fa7337b37b initial commit 2023-07-06 12:55:23 +08:00