Skip to content

Fix Parameters in Attention.h (measure_operator_cost) #1008

@KateUnger

Description

@KateUnger

in attention.h:

CostMetrics measure_operator_cost(SimEnvFactory const &sim,
                                  MultiHeadAttentionAttrs const &attrs,
                                  ParallelTensorShape const &query_shape,
                                  ParallelTensorShape const &key_shape,
                                  ParallelTensorShape const &value_shape,
                                  ProfilingSettings const &settings,
                                  MachineView const &mv);

in attention.cpp (correct):

CostMetrics measure_operator_cost(SimEnvFactory const &sim,
                                  MultiHeadAttentionAttrs const &attrs,
                                  InputParallelTensorDesc const &query_shape,
                                  InputParallelTensorDesc const &key_shape,
                                  InputParallelTensorDesc const &value_shape,
                                  ProfilingSettings const &settings,
                                  MachineView const &mv) {

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions