Skip to content

Commit 18e467b

Browse files
committed
Add ProbabilisticScorer benchmarks
1 parent 72e2295 commit 18e467b

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

lightning/src/routing/router.rs

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4984,7 +4984,7 @@ mod benches {
49844984
use chain::transaction::OutPoint;
49854985
use ln::channelmanager::{ChannelCounterparty, ChannelDetails};
49864986
use ln::features::{InitFeatures, InvoiceFeatures};
4987-
use routing::scoring::{FixedPenaltyScorer, Scorer};
4987+
use routing::scoring::{FixedPenaltyScorer, ProbabilisticScorer, ProbabilisticScoringParameters, Scorer};
49884988
use util::logger::{Logger, Record};
49894989

49904990
use test::Bencher;
@@ -5061,6 +5061,22 @@ mod benches {
50615061
generate_routes(bench, &network_graph, scorer, InvoiceFeatures::known());
50625062
}
50635063

5064+
#[bench]
5065+
fn generate_routes_with_probabilistic_scorer(bench: &mut Bencher) {
5066+
let network_graph = read_network_graph();
5067+
let params = ProbabilisticScoringParameters::default();
5068+
let scorer = ProbabilisticScorer::new(params, &network_graph);
5069+
generate_routes(bench, &network_graph, scorer, InvoiceFeatures::empty());
5070+
}
5071+
5072+
#[bench]
5073+
fn generate_mpp_routes_with_probabilistic_scorer(bench: &mut Bencher) {
5074+
let network_graph = read_network_graph();
5075+
let params = ProbabilisticScoringParameters::default();
5076+
let scorer = ProbabilisticScorer::new(params, &network_graph);
5077+
generate_routes(bench, &network_graph, scorer, InvoiceFeatures::known());
5078+
}
5079+
50645080
fn generate_routes<S: Score>(
50655081
bench: &mut Bencher, graph: &NetworkGraph, mut scorer: S, features: InvoiceFeatures
50665082
) {

0 commit comments

Comments
 (0)