ex24[gemm_grouped]: Allow to change layout/dtype (#841)

* ex24[gemm_grouped]: Allow to change layout/dtype * Address suggestion from @jackkosaian --------- Co-authored-by: danthe3rd <danthe3rd>
2023-03-01 13:13:51 +01:00
parent 92ebbf1dc4
commit f396cdd15c
2 changed files with 5 additions and 5 deletions
--- a/include/cutlass/gemm/device/gemm_universal_base.h
+++ b/include/cutlass/gemm/device/gemm_universal_base.h
@ -306,7 +306,7 @@ public:
  /// Initializes GEMM state from arguments and workspace memory
  Status initialize(
    Arguments const &args,
-    void *workspace,
+    void *workspace = nullptr,
    cudaStream_t stream = nullptr)
  {
    CUTLASS_TRACE_HOST("GemmUniversalBase::initialize() - workspace "