diff --git a/benchmarks/transport_montecarlo_single_packet_loop.py b/benchmarks/transport_montecarlo_single_packet_loop.py index 6e859b2120b..26ba7cfd121 100644 --- a/benchmarks/transport_montecarlo_single_packet_loop.py +++ b/benchmarks/transport_montecarlo_single_packet_loop.py @@ -5,6 +5,7 @@ from benchmarks.benchmark_base import BenchmarkBase from tardis.transport.montecarlo import single_packet_loop from asv_runner.benchmarks.mark import parameterize, skip_benchmark +from numba.np.ufunc.parallel import get_num_threads, get_thread_id class BenchmarkMontecarloMontecarloNumbaVpacket(BenchmarkBase): @@ -12,15 +13,15 @@ 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, self.verysimple_numba_radial_1d_geometry, self.verysimple_time_explosion, self.verysimple_opacity_state, - self.estimators, + self.transport_state.radfield_mc_estimators.create_estimator_list(get_num_threads())[get_thread_id()], self.verysimple_3vpacket_collection, self.rpacket_tracker, self.montecarlo_configuration ) +