Fix: Set StateLess() to false for assign_kernel (#2546)

4f30b78e · Niu Chong · Li Xinqi · c117d4ed · 4f30b78e
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

oneflow/core/kernel/assign_kernel.cpp oneflow/core/kernel/assign_kernel.cpp +1 -0

未找到文件。
--- 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;
 };