github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-09 14:47:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(gpu): remove carry prop after scalar mul single carry prop after scalar add/sub

Browse Source
This commit is contained in:
Agnes Leroy
2024-04-04 16:08:19 +02:00
committed by Agnès Leroy
parent daadb115aa
commit 88ff4d17cf
3 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tfhe/src/integer/gpu/server_key/radix/scalar_add.rs
View File

@@ -173,7 +173,7 @@ impl CudaServerKey {
};
self.unchecked_scalar_add_assign_async(ct, scalar, stream);
self.full_propagate_assign_async(ct, stream);
self.propagate_single_carry_assign_async(ct, stream);
}
pub fn scalar_add_assign<Scalar, T>(&self, ct: &mut T, scalar: Scalar, stream: &CudaStream)

1
tfhe/src/integer/gpu/server_key/radix/scalar_mul.rs
View File

@@ -231,7 +231,6 @@ impl CudaServerKey {
};
self.unchecked_scalar_mul_assign_async(ct, scalar, stream);
self.full_propagate_assign_async(ct, stream);
}
pub fn scalar_mul_assign<Scalar, T>(&self, ct: &mut T, scalar: Scalar, stream: &CudaStream)

2
tfhe/src/integer/gpu/server_key/radix/scalar_sub.rs
View File

@@ -152,7 +152,7 @@ impl CudaServerKey {
};
self.unchecked_scalar_sub_assign_async(ct, scalar, stream);
self.full_propagate_assign_async(ct, stream);
self.propagate_single_carry_assign_async(ct, stream);
}
pub fn scalar_sub_assign<Scalar, T>(&self, ct: &mut T, scalar: Scalar, stream: &CudaStream)