Files

.github
cmake
docs
examples
- 00_basic_gemm
- 01_cutlass_utilities
- 02_dump_reg_shmem
- 03_visualize_layout
- 04_tile_iterator
- 05_batched_gemm
- 06_splitK_gemm
- 07_volta_tensorop_gemm
- 08_turing_tensorop_gemm
- 09_turing_tensorop_conv2dfprop
- 10_planar_complex
- 11_planar_complex_array
- 12_gemm_bias_relu
- 13_two_tensor_op_fusion
- 14_ampere_tf32_tensorop_gemm
- 15_ampere_sparse_tensorop_gemm
- 16_ampere_tensorop_conv2dfprop
- 17_fprop_per_channel_bias
- 18_ampere_fp64_tensorop_affine2_gemm
- 19_tensorop_canonical
- 20_simt_canonical
- 21_quaternion_gemm
- 22_quaternion_conv
- 23_ampere_gemm_operand_reduction_fusion
- 24_gemm_grouped
- 25_ampere_fprop_mainloop_fusion
- 26_ampere_wgrad_mainloop_fusion
- 27_ampere_3xtf32_fast_accurate_tensorop_gemm
- 28_ampere_3xtf32_fast_accurate_tensorop_fprop
- 29_ampere_3xtf32_fast_accurate_tensorop_complex_gemm
- 30_wgrad_split_k
- 31_basic_syrk
- 32_basic_trmm
- 33_ampere_3xtf32_tensorop_symm
- 34_transposed_conv2d
- 35_gemm_softmax
  - CMakeLists.txt
  - epilogue_with_visitor.h
  - gemm_softmax.cu
  - gemm_with_epilogue_visitor.h
  - gemm_with_softmax.h
- 36_gather_scatter_fusion
- 40_cutlass_py
- common
- CMakeLists.txt
include
media
test
tools
.gitignore
.gitmodules
CHANGELOG.md
CMakeLists.txt
CONTRIBUTORS.md
CUDA.cmake
Doxyfile
LICENSE.txt
PUBLICATIONS.md
README.md
bin2hex.cmake
cuBLAS.cmake
cuDNN.cmake

35_gemm_softmax

CUTLASS 2.9 (NVIDIA#468 )

Apr 23, 2022

12f4108 · Apr 23, 2022

This branch is 380 commits behind NVIDIA/cutlass:main.

Name		Name	Last commit message	Last commit date
parent directory ..
CMakeLists.txt		CMakeLists.txt	CUTLASS 2.9 (NVIDIA#468 )	Apr 23, 2022
epilogue_with_visitor.h		epilogue_with_visitor.h	CUTLASS 2.9 (NVIDIA#468 )	Apr 23, 2022
gemm_softmax.cu		gemm_softmax.cu	CUTLASS 2.9 (NVIDIA#468 )	Apr 23, 2022
gemm_with_epilogue_visitor.h		gemm_with_epilogue_visitor.h	CUTLASS 2.9 (NVIDIA#468 )	Apr 23, 2022
gemm_with_softmax.h		gemm_with_softmax.h	CUTLASS 2.9 (NVIDIA#468 )	Apr 23, 2022