Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Qianfeng's avatar
    Add examples for reduction fp16/fp32/bp16/int8/fp64 for 3d/4d/5d (#342) · 14932e8d
    Qianfeng authored 
    * Update the reduce_blockwise example to support user specified data type and input+reducing dimensions

* Add examples for using reduce_multiblock_atomic_add

* Add more running examples to the default command-line

* Remove un-necessary header including

* Update to the example README.md
    14932e8d
reduce_blockwise.cpp 8.85 KB