Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
切换分支/标签
查找文件 Blame历史永久链接Permalink
  • Z
    [bf16] add bf16 kernel: scale gather sum (#39683) · 6d26b332
    由 zhangbo9674 提交于 
    * add scale gather sum

* refine CUDA_ATOMIC_WRAPPER ADD for bf16

* add gather unittest

* solve conflict

* add scale uinttest

* add sum unittest

* solve conflict

* refine gather unittest

* refine unittest
    6d26b332
test_gather_op.py 12.3 KB