From 5a860d407a7f47e3c8b9586f8630a796181b72d9 Mon Sep 17 00:00:00 2001
From: Maths <644311798@qq.com>
Date: Fri, 15 Dec 2023 19:09:45 +0800
Subject: [PATCH] Update arg_kv.h

---
 yacl/utils/spi/argument/arg_kv.h | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/yacl/utils/spi/argument/arg_kv.h b/yacl/utils/spi/argument/arg_kv.h
index 100a7b98..52302efc 100644
--- a/yacl/utils/spi/argument/arg_kv.h
+++ b/yacl/utils/spi/argument/arg_kv.h
@@ -39,14 +39,14 @@ class SpiArg {
   }
 
   template <typename T>
-  &SpiArg operator=(const T &value) {
+  SpiArg& operator=(const T &value) {
     value_ = value;
     return *this;
   }
 
   // Specialized functions of operator=<T>
-  SpiArg operator=(const char *value);
-  SpiArg operator=(const std::string &value);
+  SpiArg& operator=(const char *value);
+  SpiArg& operator=(const std::string &value);
 
   const std::string &Key() const;
   bool HasValue() const;