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

WIP: try inline never on hlapi bench functions

Browse Source
This commit is contained in:
David Testé
2026-01-16 11:09:28 +01:00
parent 5e0a2ee366
commit c047ee700d
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@@ -1639,7 +1639,7 @@ bench_hlapi: install_rs_check_toolchain
RUSTFLAGS="$(RUSTFLAGS)" __TFHE_RS_BENCH_BIT_SIZES_SET=$(BIT_SIZES_SET) \
cargo $(CARGO_RS_CHECK_TOOLCHAIN) bench \
--bench hlapi \
--features=integer,internal-keycache,pbs-stats -p tfhe-benchmark --
--features=integer,internal-keycache,pbs-stats -p tfhe-benchmark -- '::add|::mul|::bitand'
.PHONY: bench_hlapi_gpu # Run benchmarks for integer operations on GPU
bench_hlapi_gpu: install_rs_check_toolchain

6
tfhe-benchmark/benches/high_level_api/bench.rs
View File

@@ -330,6 +330,7 @@ where
}
}
#[inline(never)]
fn bench_fhe_type_op<FheType, Op>(
c: &mut Criterion,
client_key: &ClientKey,
@@ -383,6 +384,7 @@ fn bench_fhe_type_op<FheType, Op>(
macro_rules! bench_type_binary_op {
(type_name: $fhe_type:ident, right_type_name: $fhe_right_type:ident,left_type: $left_type:ty, right_type: $right_type:ty, display_name: $display_name:literal, operation: $op:ident) => {
::paste::paste! {
#[inline(never)]
fn [<bench_ $fhe_type:snake _ $op>](c: &mut Criterion, cks: &ClientKey) {
bench_fhe_type_op(
c,
@@ -413,6 +415,7 @@ macro_rules! bench_type_binary_scalar_op {
rng: $rng_fn:expr
) => {
::paste::paste! {
#[inline(never)]
fn [<bench_ $fhe_type:snake _scalar_ $op>](c: &mut Criterion, cks: &ClientKey) {
bench_fhe_type_op(
c,
@@ -435,6 +438,7 @@ macro_rules! bench_type_binary_scalar_op {
macro_rules! bench_type_unary_op {
(type_name: $fhe_type:ident, integer_type: $integer_type:ty, display_name: $display_name:literal, operation: $op:ident) => {
::paste::paste! {
#[inline(never)]
fn [<bench_ $fhe_type:snake _ $op>](c: &mut Criterion, cks: &ClientKey) {
bench_fhe_type_op(
c,
@@ -456,6 +460,7 @@ macro_rules! bench_type_unary_op {
macro_rules! bench_type_ternary_op {
(type_name: $fhe_type:ident, integer_type: $integer_type:ty, display_name: $display_name:literal, operation: $op:ident) => {
::paste::paste! {
#[inline(never)]
fn [<bench_ $fhe_type:snake _ $op>](c: &mut Criterion, cks: &ClientKey) {
bench_fhe_type_op(
c,
@@ -477,6 +482,7 @@ macro_rules! bench_type_ternary_op {
macro_rules! bench_type_array_op {
(type_name: $fhe_type:ident, integer_type: $integer_type:ty, display_name: $display_name:literal, operation: $op:ident) => {
::paste::paste! {
#[inline(never)]
fn [<bench_ $fhe_type:snake _ $op>](c: &mut Criterion, cks: &ClientKey) {
bench_fhe_type_op(
c,