From 4f30b78e575fcfbd9d8056038f1debdaccb1f3c6 Mon Sep 17 00:00:00 2001
From: Niu Chong <niuchong893184@gmail.com>
Date: Tue, 7 Jan 2020 17:23:59 +0800
Subject: [PATCH] Fix: Set StateLess() to false for assign_kernel (#2546)

---
 oneflow/core/kernel/assign_kernel.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/oneflow/core/kernel/assign_kernel.cpp b/oneflow/core/kernel/assign_kernel.cpp
index beaa23a6f4..aabf9d2e5f 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;
 };
-- 
GitLab