call cudnnBatchNormalizationForwardInference if trainable == flase (#1197)

Former-commit-id: a21dea46

call cudnnBatchNormalizationForwardInference if trainable == flase (#1197)
Former-commit-id: a21dea46
d5c6eecb · Li Xinqi · Jinhui Yuan · 5fa70913 · d5c6eecb
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

oneflow/core/kernel/normalization_kernel.cpp oneflow/core/kernel/normalization_kernel.cpp +1 -1

未找到文件。
--- a/oneflow/core/kernel/normalization_kernel.cpp
+++ b/oneflow/core/kernel/normalization_kernel.cpp
@@ -92,7 +92,7 @@ void NormalizationKernel<DeviceType::kGPU, float>::NormalizationCudnnForward(
  float* moving_mean = BnInOp2Blob("moving_mean")->mut_dptr<float>();
  float* moving_variance = BnInOp2Blob("moving_variance")->mut_dptr<float>();
  double epsilon = this->op_conf().normalization_conf().epsilon();
-  if (Global<JobDesc>::Get()->IsTrain()) {
+  if (this->op_conf().trainable()) {
    InitMovingMeanAndMovingVariance(ctx, BnInOp2Blob, false);
    double momentum = this->op_conf().normalization_conf().momentum();
    CudaCheck(cudnnBatchNormalizationForwardTraining(