struct dnnl::ukernel::brgemm_pack_b#

Overview#

struct brgemm_pack_b: public dnnl::handle
{
    // construction

    brgemm_pack_b();

    brgemm_pack_b(
        memory::dim K,
        memory::dim N,
        memory::dim in_ld,
        memory::dim out_ld,
        memory::data_type in_dt,
        memory::data_type out_dt,
        bool allow_empty = false
        );

    // methods

    bool need_pack() const;
    void generate();
    void execute(const void* in, void* out) const;
};

Inherited Members#

public:
    // methods

    handle<T, traits>& operator = (const handle<T, traits>&);
    handle<T, traits>& operator = (handle<T, traits>&&);
    void reset(T t, bool weak = false);
    T get(bool allow_empty = false) const;
    operator T () const;
    operator bool () const;
    bool operator == (const handle<T, traits>& other) const;
    bool operator != (const handle& other) const;

Detailed Documentation#

Construction#

brgemm_pack_b()

Default constructor. Produces an empty object.

brgemm_pack_b(
    memory::dim K,
    memory::dim N,
    memory::dim in_ld,
    memory::dim out_ld,
    memory::data_type in_dt,
    memory::data_type out_dt,
    bool allow_empty = false
    )

Constructs a BRGeMM ukernel packing tensor B object.

Parameters:

K	Dimension K.
N	Dimension N.
in_ld	Input leading dimension.
out_ld	Output leading dimension. Specifies a block by N dimension during data packing.
in_dt	Input data type.
out_dt	Output data type.
allow_empty	A flag signifying whether construction is allowed to fail without throwing an exception. In this case an empty object will be produced. This flag is optional and defaults to false.

Methods#

bool need_pack() const

Returns the flag if packing is expected by BRGeMM ukernel kernel.

void generate()

Generates an executable part of BRGeMM ukernel packing B object.

void execute(const void* in, void* out) const

Executes a BRGeMM ukernel packing tensor B object.