add kind to cipher

Signed-off-by: sagewe <wbwmat@gmail.com>

python/fate/arch/context/_cipher.py

@@ -72,23 +72,23 @@ def setup(self, options: typing.Optional[dict] = None):
 
             sk, pk, coder = keygen(key_size)
             tensor_cipher = PHETensorCipher.from_raw_cipher(pk, coder, sk, evaluator)
-            return PHECipher(key_size, pk, sk, evaluator, coder, tensor_cipher, True, True, True)
+            return PHECipher(kind, key_size, pk, sk, evaluator, coder, tensor_cipher, True, True, True)
 
         if kind == "ou":
             from fate.arch.protocol.phe.ou import evaluator, keygen
             from fate.arch.tensor.phe import PHETensorCipher
 
             sk, pk, coder = keygen(key_size)
             tensor_cipher = PHETensorCipher.from_raw_cipher(pk, coder, sk, evaluator)
-            return PHECipher(key_size, pk, sk, evaluator, coder, tensor_cipher, False, False, True)
+            return PHECipher(kind, key_size, pk, sk, evaluator, coder, tensor_cipher, False, False, True)
 
         elif kind == "mock":
             from fate.arch.protocol.phe.mock import evaluator, keygen
             from fate.arch.tensor.phe import PHETensorCipher
 
             sk, pk, coder = keygen(key_size)
             tensor_cipher = PHETensorCipher.from_raw_cipher(pk, coder, sk, evaluator)
-            return PHECipher(key_size, pk, sk, evaluator, coder, tensor_cipher, True, False, False)
+            return PHECipher(kind, key_size, pk, sk, evaluator, coder, tensor_cipher, True, False, False)
 
         else:
             raise ValueError(f"Unknown PHE keygen kind: {self.kind}")
@@ -97,6 +97,7 @@ def setup(self, options: typing.Optional[dict] = None):
 class PHECipher:
     def __init__(
         self,
+        kind,
         key_size,
         pk,
         sk,
@@ -107,6 +108,7 @@ def __init__(
         can_support_squeeze,
         can_support_pack,
     ) -> None:
+        self._kind = kind
         self._key_size = key_size
         self._pk = pk
         self._sk = sk
@@ -117,17 +119,21 @@ def __init__(
         self._can_support_squeeze = can_support_squeeze
         self._can_support_pack = can_support_pack
 
+    @property
+    def kind(self):
+        return self._kind
+
     @property
     def can_support_negative_number(self):
-        return self._tensor_cipher.can_support_negative_number
+        return self._can_support_negative_number
 
     @property
     def can_support_squeeze(self):
-        return self._tensor_cipher.can_support_squeeze
+        return self._can_support_squeeze
 
     @property
     def can_support_pack(self):
-        return self._tensor_cipher.can_support_pack
+        return self._can_support_pack
 
     @property
     def key_size(self):