diff --git a/oneflow/core/kernel/sink_tick_kernel.h b/oneflow/core/kernel/sink_tick_kernel.h index d20a6707b89a7e4809e5ef086ef31fcca44675a1..c4c25970ac9dd06d861a2ddf4be0d6f0d0941d0a 100644 --- a/oneflow/core/kernel/sink_tick_kernel.h +++ b/oneflow/core/kernel/sink_tick_kernel.h @@ -29,9 +29,7 @@ class SinkTickKernel final : public KernelIf<DeviceType::kCPU> { private: void ForwardDataContent(const KernelCtx& ctx, std::function<Blob*(const std::string&)> BnInOp2Blob) const override {} - const PbMessage& GetCustomizedOpConf() const override { - return this->op_conf().source_tick_conf(); - } + const PbMessage& GetCustomizedOpConf() const override { return this->op_conf().sink_tick_conf(); } }; } // namespace oneflow