InfiniTensor/test/kernels
xgqdut2016 d3e7543291
Cuda softmax (#129)
* "add softmax.cu,.cc,.h"

* Modify cuda softmax

* "modified the introduction of softmax.cu"

* "add format of cuda_softmax.h"

* "modified where.cc(.cu,.h) and softmax.cu"

* "modified format"

* Fix cpu softmax kernel

* "modified the // introduction of softmax.cu"

* "modified softmax.cu and use 1D block"

* "modified softmax.cu,format, and use 1D block"

* "introduce share mem to speed softmax"

* "reduce the input of function"

* modified the format

* remodify 2D block softmax

* remodify 1D block softmax

* modified the share memory

* add warp reduce

* conflict solve two

* remove extra space line

* solve comment

---------

Co-authored-by: Haojie Wang <haojie0429@gmail.com>
Co-authored-by: panzezhong <panzezhong@qiyuanlab.com>
2023-11-06 08:56:23 +08:00
..
bang Pooling ceil mode (#155) 2023-10-09 20:51:39 +08:00
cuda Cuda softmax (#129) 2023-11-06 08:56:23 +08:00
intelcpu memory_allocator (#103) 2023-08-13 13:39:35 +08:00
kunlun Xpu task4 support: add softmax (#172) 2023-10-30 16:01:05 +08:00
nativecpu add transpose, concat and split for native cpu (#158) 2023-10-12 10:14:28 +08:00