Contents

class

SushiBLAS::Dispatch::GemmEpilogue

template <typename Epilogue>

Holds a GEMM epilogue functor together with the regions it reads and writes.

Declared in
include/SushiBLAS/engine/blas/dispatch.hpp

Template parameters

Epilogue

The functor type, under the contract execute_gemm states for a bare one.

See also

include/SushiBLAS/engine/blas/README.md

Public member functions

explicit GemmEpilogue(Epilogue functor)

Wraps functor, declaring nothing yet.

GemmEpilogue & reads(const Tensor &operand)

Declares that the epilogue reads operand.

Returns

*this, so declarations chain.

Exceptions

std::runtime_error

if operand has no storage.

GemmEpilogue & writes(const Tensor &operand)

Declares that the epilogue writes operand.

Returns

*this, so declarations chain.

Exceptions

std::runtime_error

if operand has no storage.

See also

include/SushiBLAS/engine/blas/README.md

const Epilogue & functor() const noexcept

Returns the functor itself, the only part that reaches the device.

const OperandSet & operands() const noexcept

Returns the tensors declared through reads() and writes().

const std::vector< SushiRuntime::Core::ResourceRegion > & read_regions() const noexcept

Returns the regions declared through reads().

const std::vector< SushiRuntime::Core::ResourceRegion > & write_regions() const noexcept

Returns the regions declared through writes().