Epilogue with RELU (#86)

codeplaysoftware · Jul 16, 2024 · fae9b2e · fae9b2e
1 parent 820792e
commit fae9b2e
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
@@ -31,7 +31,6 @@
 
 #define CUTLASS_SYCLCOMPAT_PROFILING_ENABLED
 
-#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"