NVIDIA/cutlass
1
0
Fork 0
mirror of https://github.com/NVIDIA/cutlass.git synced 2026-05-25 07:14:30 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f78994bb400ccc29ad3bf7d06a41641859e14557
cutlass/include
History
masahi dceabd4c5a Support half precision sigmoid activation (#378) 
* Support half precision sigmoid activation

* introduce a vectorized variant using fast_tanh

* move the math to fast_math.h

* fixed compile

* .raw() -> .to_half()

Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
2021-12-22 14:45:06 -05:00
..
cutlass
Support half precision sigmoid activation (#378)
2021-12-22 14:45:06 -05:00