squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
e7a61c761a
cutlass/examples/35_gemm_softmax
History
Yujia Zhai 04a9777b87
Softmax (#546) 
* add test layernorm g-mem version

* Delete include/configure directory

* Delete examples/test_layernorm directory

* Update gemm_with_softmax.h

* Update gemm_softmax.cu

* Update linear_combination.h

* Update fast_math.h

* remove redundant vars

Co-authored-by: yujia.zhai <yujia.zhai@bytedance.com>
Co-authored-by: yuzhai <yuzhai@nvidia.com>
2022-07-02 01:19:18 -04:00
..
CMakeLists.txt CUTLASS 2.9 (#468) 2022-04-23 15:02:38 -04:00
epilogue_with_visitor.h CUTLASS 2.9 (#468) 2022-04-23 15:02:38 -04:00
gemm_softmax.cu Softmax (#546) 2022-07-02 01:19:18 -04:00
gemm_with_epilogue_visitor.h CUTLASS 2.9 (#468) 2022-04-23 15:02:38 -04:00
gemm_with_softmax.h Softmax (#546) 2022-07-02 01:19:18 -04:00