diff --git a/oneflow/core/kernel/assign_kernel.cpp b/oneflow/core/kernel/assign_kernel.cpp
index beaa23a6f48969dc9ce846b3e75f5db2bcfe2b54..aabf9d2e5f8000ef53a652b2fd3776a24cde7157 100644
--- a/oneflow/core/kernel/assign_kernel.cpp
+++ b/oneflow/core/kernel/assign_kernel.cpp
@@ -10,6 +10,7 @@ class AssignKernel final : public KernelIf<device_type> {
   ~AssignKernel() override = default;
 
  private:
+  bool IsStateless() const override { return false; }
   void ForwardDataContent(const KernelCtx&,
                           std::function<Blob*(const std::string&)>) const override;
 };