Skip to content

Commit

Permalink
fix benchmark single_packet_loop
Browse files Browse the repository at this point in the history
Signed-off-by: Asish Kumar <[email protected]>
  • Loading branch information
officialasishkumar committed Jul 11, 2024
1 parent ea6175a commit 5bae9d4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions benchmarks/transport_montecarlo_single_packet_loop.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,23 +4,23 @@

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):
"""
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
)

0 comments on commit 5bae9d4

Please sign in to comment.