softmax with EVT (draft)

codeplaysoftware · Jan 7, 2025 · bf7c5e0 · bf7c5e0
1 parent 175f05b
commit bf7c5e0
Show file tree

Hide file tree

Showing 6 changed files with 1,061 additions and 3 deletions.
diff --git a/examples/sycl/pvc/CMakeLists.txt b/examples/sycl/pvc/CMakeLists.txt
@@ -42,6 +42,11 @@ cutlass_example_add_executable(
   pvc_gemm_with_epilogue_gelu.cpp
 )
 
+cutlass_example_add_executable(
+  pvc_gemm_with_epilogue_softmax
+  pvc_gemm_with_epilogue_softmax.cpp
+)
+
 cutlass_example_add_executable(
   pvc_collective_builder
   pvc_collective_builder.cpp