From 7c8c362365a9a4fc9ed0c53416d6b597e15d7a7f Mon Sep 17 00:00:00 2001 From: Asish Kumar Date: Tue, 9 Jul 2024 18:27:04 +0530 Subject: [PATCH] benchmark single packet loop Signed-off-by: Asish Kumar --- benchmarks/benchmark_base.py | 5 +++-- benchmarks/transport_montecarlo_single_packet_loop.py | 4 +--- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/benchmarks/benchmark_base.py b/benchmarks/benchmark_base.py index 4fc87332353..4136fd8b6eb 100644 --- a/benchmarks/benchmark_base.py +++ b/benchmarks/benchmark_base.py @@ -24,6 +24,7 @@ from tardis.transport.montecarlo.packet_collections import ( VPacketCollection, ) +from tardis.transport.montecarlo.packet_trackers import RPacketTracker class BenchmarkBase: @@ -361,8 +362,8 @@ def montecarlo_configuration(self): return montecarlo_configuration.MonteCarloConfiguration() @property - def rpacket_tracker(self): - return self.simulation_rpacket_tracking_enabled.transport.transport_state.rpacket_tracker + def rpacket_tracker(self): + return RPacketTracker(0) @property def transport_state(self): diff --git a/benchmarks/transport_montecarlo_single_packet_loop.py b/benchmarks/transport_montecarlo_single_packet_loop.py index a92418e40d2..4da3961d986 100644 --- a/benchmarks/transport_montecarlo_single_packet_loop.py +++ b/benchmarks/transport_montecarlo_single_packet_loop.py @@ -4,7 +4,6 @@ from benchmarks.benchmark_base import BenchmarkBase from tardis.transport.montecarlo import single_packet_loop -from asv_runner.benchmarks.mark import skip_benchmark class BenchmarkMontecarloMontecarloNumbaVpacket(BenchmarkBase): @@ -12,7 +11,6 @@ class BenchmarkMontecarloMontecarloNumbaVpacket(BenchmarkBase): Class to benchmark the single packet loop function. """ - @skip_benchmark def time_single_packet_loop(self): single_packet_loop.single_packet_loop( self.packet, @@ -20,7 +18,7 @@ def time_single_packet_loop(self): self.verysimple_time_explosion, self.verysimple_opacity_state, self.estimators, - self.verysimple_packet_collection, + self.verysimple_3vpacket_collection, self.rpacket_tracker, self.montecarlo_configuration )