Commit a124067c authored by fsx950223's avatar fsx950223
Browse files

fix name

parent 87df7683
......@@ -59,7 +59,7 @@ struct DeviceSparseEmbeddingsForwardLayernorm : public BaseOperator
EmbeddingDim_(EmbeddingDim),
IndexLength_(IndexLength),
epsilon_(epsilon),
reduce_op_(elementwise_op)
elementwise_op_(elementwise_op)
{
grid_size_ = (IndexLength + DimClusterSize - 1) / DimClusterSize;
}
......@@ -72,7 +72,7 @@ struct DeviceSparseEmbeddingsForwardLayernorm : public BaseOperator
ck::index_t EmbeddingDim_;
ck::index_t IndexLength_;
AccDataType epsilon_;
ElementwiseOperation reduce_op_;
ElementwiseOperation elementwise_op_;
size_t grid_size_;
};
......@@ -146,7 +146,7 @@ struct DeviceSparseEmbeddingsForwardLayernorm : public BaseOperator
arg.p_beta_,
out_desc,
arg.epsilon_,
arg.reduce_op_);
arg.elementwise_op_);
return (avg_time);
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment