From 6c891db9f695d158c57b3744640e9d91335283d3 Mon Sep 17 00:00:00 2001 From: Wenbo Yang Date: Thu, 31 Jul 2025 10:12:53 +0800 Subject: [PATCH] Fix epilogue::thread::Convert cannot be used with cute::collective::DefaultEpilogue. (#2333) --- include/cutlass/epilogue/thread/conversion_op.h | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/include/cutlass/epilogue/thread/conversion_op.h b/include/cutlass/epilogue/thread/conversion_op.h index 432906ac..19bbc03a 100644 --- a/include/cutlass/epilogue/thread/conversion_op.h +++ b/include/cutlass/epilogue/thread/conversion_op.h @@ -62,6 +62,7 @@ public: using ElementOutput = ElementOutput_; using ElementAccumulator = ElementAccumulator_; using ElementCompute = ElementAccumulator_; + using ElementD = ElementOutput; // for use with cute::collective::DefaultEpilogue static int const kCount = Count; @@ -123,6 +124,21 @@ public: return destination_converter(accumulator); } + + // + // Specializations for scalar (for use with cute::collective::DefaultEpilogue) + // + CUTLASS_HOST_DEVICE + ElementD operator()(ElementAccumulator const accumulator, ElementAccumulator const source) const { + NumericConverter destination_converter; + return destination_converter(source); + } + + CUTLASS_HOST_DEVICE + ElementD operator()(ElementAccumulator const accumulator) const { + NumericConverter destination_converter; + return destination_converter(accumulator); + } }; /////////////////////////////////////////////////////////////////////////////////////////////////