@@ -4984,7 +4984,7 @@ mod benches {
4984
4984
use chain:: transaction:: OutPoint ;
4985
4985
use ln:: channelmanager:: { ChannelCounterparty , ChannelDetails } ;
4986
4986
use ln:: features:: { InitFeatures , InvoiceFeatures } ;
4987
- use routing:: scoring:: { FixedPenaltyScorer , Scorer } ;
4987
+ use routing:: scoring:: { FixedPenaltyScorer , ProbabilisticScorer , ProbabilisticScoringParameters , Scorer } ;
4988
4988
use util:: logger:: { Logger , Record } ;
4989
4989
4990
4990
use test:: Bencher ;
@@ -5061,6 +5061,22 @@ mod benches {
5061
5061
generate_routes ( bench, & network_graph, scorer, InvoiceFeatures :: known ( ) ) ;
5062
5062
}
5063
5063
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
+
5064
5080
fn generate_routes < S : Score > (
5065
5081
bench : & mut Bencher , graph : & NetworkGraph , mut scorer : S , features : InvoiceFeatures
5066
5082
) {
0 commit comments