Epilogue with RELU

codeplaysoftware · Jul 16, 2024 · 5ac4634 · 5ac4634
1 parent 8f3bd67
commit 5ac4634
Show file tree

Hide file tree

Showing 6 changed files with 493 additions and 6 deletions.
diff --git a/examples/sycl/pvc/CMakeLists.txt b/examples/sycl/pvc/CMakeLists.txt
@@ -28,6 +28,11 @@
 
 
 cutlass_example_add_executable(
-  pvc_bfloat_dpas_gemm_cute
-  pvc_bfloat_dpas_gemm_cute.cpp
+  pvc_gemm
+  pvc_gemm.cpp
+)
+
+cutlass_example_add_executable(
+  pvc_gemm_with_epilogue_relu
+  pvc_gemm_with_epilogue_relu.cpp
 )
diff --git a/...es/sycl/pvc/pvc_bfloat_dpas_gemm_cute.cpp → examples/sycl/pvc/pvc_gemm.cpp b/...es/sycl/pvc/pvc_bfloat_dpas_gemm_cute.cpp → examples/sycl/pvc/pvc_gemm.cpp
@@ -29,7 +29,6 @@
  *
  **************************************************************************************************/
 
-#include "cutlass/gemm/device/gemm.h"
 #include "cutlass/epilogue/collective/default_epilogue.hpp"
 #include "cutlass/epilogue/collective/intel_pvc_epilogue.hpp"
 #include "cutlass/epilogue/fusion/intel_pvc_callbacks.hpp"