chore(gpu): fix bad merge

zama-ai · Sep 5, 2024 · c84d0b6 · c84d0b6
1 parent d454b53
commit c84d0b6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 12 deletions.
diff --git a/tfhe/src/high_level_api/integers/signed/base.rs b/tfhe/src/high_level_api/integers/signed/base.rs
@@ -297,7 +297,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -341,7 +341,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -385,7 +385,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -429,7 +429,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -473,7 +473,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -526,7 +526,10 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                (crate::FheUint32::new(result), FheBool::new(is_ok))
+                (
+                    crate::FheUint32::new(result, cuda_key.tag.clone()),
+                    FheBool::new(is_ok, cuda_key.tag.clone()),
+                )
             }),
         })
     }

diff --git a/tfhe/src/high_level_api/integers/unsigned/base.rs b/tfhe/src/high_level_api/integers/unsigned/base.rs
@@ -396,7 +396,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -440,7 +440,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -484,7 +484,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -528,7 +528,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -572,7 +572,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -625,7 +625,10 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                (super::FheUint32::new(result), FheBool::new(is_ok))
+                (
+                    super::FheUint32::new(result, cuda_key.tag.clone()),
+                    FheBool::new(is_ok, cuda_key.tag.clone()),
+                )
             }),
         })
     }