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

chore(hpu): make pcc happy

Browse Source
This commit is contained in:
pgardratzama
2025-12-29 12:48:44 +01:00
parent b0d7727f9a
commit 499f210407
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backends/tfhe-hpu-backend/src/fw/fw_impl/llt/mod.rs
View File

@@ -239,7 +239,12 @@ pub fn iop_erc_20(prog: &mut Program) {
pub fn iop_erc_20_simd(prog: &mut Program) { pub fn iop_erc_20_simd(prog: &mut Program) {
// Add Comment header // Add Comment header
prog.push_comment("ERC_20_SIMD (new_from, new_to) <- (from, to, amount)".to_string()); prog.push_comment("ERC_20_SIMD (new_from, new_to) <- (from, to, amount)".to_string());
simd(prog, crate::asm::iop::SIMD_N, fw_impl::llt::iop_erc_20_rtl, None); simd(
prog,
crate::asm::iop::SIMD_N,
fw_impl::llt::iop_erc_20_rtl,
None,
);
} }
#[instrument(level = "trace", skip(prog))] #[instrument(level = "trace", skip(prog))]
@@ -430,7 +435,8 @@ pub fn iop_erc_20_rtl(prog: &mut Program, batch_index: u8, kogge_blk_w: Option<u
if let Some(blk_w) = kogge_blk_w { if let Some(blk_w) = kogge_blk_w {
kogge::add(prog, dst_to, src_to, src_amount.clone(), None, blk_w) kogge::add(prog, dst_to, src_to, src_amount.clone(), None, blk_w)
+ kogge::sub(prog, dst_from, src_from, src_amount, blk_w) + kogge::sub(prog, dst_from, src_from, src_amount, blk_w)
} else { // Default to ripple carry } else {
// Default to ripple carry
kogge::ripple_add(dst_to, src_to, src_amount.clone(), None) kogge::ripple_add(dst_to, src_to, src_amount.clone(), None)
+ kogge::ripple_sub(prog, dst_from, src_from, src_amount) + kogge::ripple_sub(prog, dst_from, src_from, src_amount)
} }

6
tfhe/src/high_level_api/prelude.rs
View File

@@ -9,9 +9,9 @@
pub use crate::high_level_api::traits::{ pub use crate::high_level_api::traits::{
BitSlice, CiphertextList, DivRem, FheDecrypt, FheEncrypt, FheEq, FheKeyswitch, FheMax, FheMin, BitSlice, CiphertextList, DivRem, FheDecrypt, FheEncrypt, FheEq, FheKeyswitch, FheMax, FheMin,
FheOrd, FheTrivialEncrypt, FheTryEncrypt, FheTryTrivialEncrypt, FheWait, Flip, IfThenElse, FheOrd, FheTrivialEncrypt, FheTryEncrypt, FheTryTrivialEncrypt, FheWait, Flip, IfThenElse,
OverflowingAdd, OverflowingMul, OverflowingNeg, OverflowingSub, ReRandomize, RotateLeft, IfThenZero, OverflowingAdd, OverflowingMul, OverflowingNeg, OverflowingSub, ReRandomize,
RotateLeftAssign, RotateRight, RotateRightAssign, ScalarIfThenElse, SquashNoise, Tagged, RotateLeft, RotateLeftAssign, RotateRight, RotateRightAssign, ScalarIfThenElse, SquashNoise,
IfThenZero, Tagged,
}; };
#[cfg(feature = "hpu")] #[cfg(feature = "hpu")]
pub use crate::high_level_api::traits::{FheHpu, HpuHandle}; pub use crate::high_level_api::traits::{FheHpu, HpuHandle};