Skip to content

Commit 6242ae5

Browse files
fix the typo in IPEXBatchNormOp backward RECORD_FUNCTION (#149)
1 parent 5d92d38 commit 6242ae5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch_ipex/csrc/cpu/BatchNorm.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ torch::autograd::variable_list
163163
IPEXBatchNormOp::backward(torch::autograd::AutogradContext *ctx,
164164
torch::autograd::variable_list grad_outputs) {
165165
#if defined(IPEX_PROFILE_OP)
166-
RECORD_FUNCTION("IPEXConvolutionOp::backward", std::vector<c10::IValue>({}));
166+
RECORD_FUNCTION("IPEXBatchNormOp::backward", std::vector<c10::IValue>({}));
167167
#endif
168168
auto train = ctx->saved_data["train"].toBool();
169169
auto eps = ctx->saved_data["eps"].toDouble();

0 commit comments

Comments
 (0)