-
Notifications
You must be signed in to change notification settings - Fork 250
Fix Parameters in Attention.h (measure_operator_cost) #1008
Copy link
Copy link
Open
Milestone
Description
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) {Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels