From 2cc2c7ba1f2f9a3be97411c211bcf9c8e75178b8 Mon Sep 17 00:00:00 2001 From: alexfreudenberg <50404848+alexfreudenberg@users.noreply.github.com> Date: Wed, 14 Sep 2022 04:34:20 +0200 Subject: [PATCH] Add set_k_partition function (#624) A member function set_k_partition is required for the instatiation of cutlass::gemm::kernel::Gemm, even though SplitKSerial is false --- include/cutlass/epilogue/thread/conversion_op.h | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/include/cutlass/epilogue/thread/conversion_op.h b/include/cutlass/epilogue/thread/conversion_op.h index baf9eb7d..ba3738ed 100644 --- a/include/cutlass/epilogue/thread/conversion_op.h +++ b/include/cutlass/epilogue/thread/conversion_op.h @@ -90,6 +90,12 @@ public: CUTLASS_HOST_DEVICE Convert(Params const ¶ms = Params()) { + } + + /// Functionally required for serial reduction in the epilogue + CUTLASS_HOST_DEVICE + void set_k_partition(int k_partition, int k_partition_count) { + } /// Returns true if source is needed based on state of runtime arguments