XENONnT / fuse

73%

LAST BUILD ON BRANCH main

branch: main

CHANGE BRANCH

x

Reset

Sync Branches

• main
• 1.5.1
• 199-negative-pmt-channel
• Increase_test_timeout
• add_PR_template
• add_back_python3.9
• add_fuse_input_dtype
• additional_tests
• args_refactoring
• automatic_documentation_pages
• beta_yields
• bump-1.6.0
• bump-v1.6.1
• bump-v1.6.2
• bump-v1.6.3
• bump-v156
• bump_v144
• bump_v153
• bump_v154
• bump_v155
• chunking
• clean-contexts
• cleanup-unused-configs
• clip-fdc-map
• cluster-dtype-int16
• cmake
• config_cleanup
• continuous_pulse_id
• cosmetics
• cs2_w_bias_xy
• cs2_w_bias_xy_segee_pi_relcy
• csv_generator_fix
• csv_input_update
• csv_timing_fix
• cuts
• daq_photon_sorting
• dark_counts
• data_type_csv_input
• debug
• debug_cut_plugins
• description_tagged_clusters
• dev_myoshida
• dummy_map_option
• early_exit_PMTResponseAndDAQ
• electron_drift_fix
• elife-cs2
• embellishing
• event_id_bugfix
• event_truth_pos
• exciton_fix_v2
• exciton_ratio
• extend_event_truth
• faster_lineage
• fastsim_in_fuse
• field_dtype_fix
• fix-chunk-ends-late
• fix-context
• fix-lineage-cluster-ids
• fix/get-parent-same-time-tag
• fix_data_kind_records_truth
• fix_documentation_bug
• fix_noise_offset
• fix_numpy_version
• fix_raw_area_calculation
• fix_rechunking_microphysicssummary
• fix_s2_photon_timing_bug
• fix_write_run_id_to_config
• fix_xyz_pri
• fixed_timing
• fixed_timing_fix
• flag_for_pmt_ap
• fuse-straxen-compatibility
• fuse_context
• import_mongo_from_utilix
• in_one_line
• increase-g4-string-fileds
• input_dtype
• issue145
• itp_maps_check
• keep-nest-oustide-validity
• lce-pattern-map-change
• less_cut_delayed
• lineage_clustering
• lineage_trackid
• load_selected_event_ragne
• logging_fix
• low_strax
• made_in_heaven
• mc-override
• mem_eff_input
• migdal_yield
• minor_updates
• more_afterpulses
• more_doc
• more_tests
• neutron_Transportation
• new_release
• no-corrected-aft
• no_requires_python
• no_s1_photon_time_clipping
• nr_micro_clustering
• nr_sd_override
• nr_sd_override_step
• only_part_of_pi
• optimize_merge_lineage
• option_to_disable_noise
• optional_aft_scaling
• optmize_cluster_and_classify
• overlap-peak-truth
• override_corrections
• patch-pre-commit-docformatter
• pattern-map-interpolation-config
• peak_truth
• per_electron_simulation
• performance
• phd
• photoNuclear_primary_neutron
• photon_finding
• pi-issue-patch
• pr/lineage-clustering
• pr/pmt-afterpulses
• pre-commit-ci-update-config
• pre_commit_debris
• pre_commit_fix
• preferred-interpolation-method
• public_config
• pulse_id_dtype_fix
• ramirezdiego-patch-1
• random_seed_from_user
• raw-records-save-when
• release
• release_1.2.1
• release_1.2.3
• release_1.4.0
• release_1.4.1
• release_1.4.3
• release_1.5.0
• remove_cmt_urls
• remove_wfsim_connection_plugin
• rename_S1PhotonHits_output
• rm_filter
• rm_unused
• s2_aft
• s2_width
• save_t_microphysics
• simple_test
• skip-aft-correction
• small_improvements
• sort_clustered_interactions
• sort_geant4_interactions
• spe_cdf_fix
• sprinkle
• stable_sort
• str_output_folder
• strax_2_straxen_3
• time_dtype_fix
• track-all-configs
• truth_data
• typo-fix
• unify_graphs
• unlock_seed_in_yields
• update-docs-and-notebooks
• update_documentation
• update_help_source_rate
• update_logging
• update_nr_flag
• update_nr_yields
• use_pre-commit
• use_straxen_docs_utils
• v1.5.2
• write_run_id_to_config
• yield_parameters_from_config
• yields-no-rf
• yuema137-patch-csvinput

Build # 28634797171

Build Type

push

github

Committed by web-flow

Commit Message

perf(pmt_afterpulses): two-pass numba scheme for photon_afterpulse (#385)

* perf(pmt_afterpulses): two-pass numba scheme for photon_afterpulse

Replace the per-element Python orchestration inside `photon_afterpulse`
(per-element numpy ops: scale rU0, fancy-index prob_ap, compute selection
mask, gather selected channels, argmin-based inverse CDF) with a two-pass
numba scheme:

  Pass 1 — `_ap_select_kernel`: one tight loop over N rows that scales
    rU0 in place by the per-element modifier (halving the DPE entries),
    compares against the per-channel probability ceiling, and emits the
    indices that pass the selection plus the running max for the > 0.5
    warning check. Eliminates the per-element numpy intermediates.

  Pass 2 — `_ap_kernel_nonuniform`: two register-resident O(K) scans per
    selected photon (one over the delay-time CDF, one over the amplitude
    CDF). Replaces the broadcast-+-argmin pattern that materialised an
    (N_sel, K) intermediate per call. The amplitude CDF is sometimes 1D
    and sometimes 2D — both shapes are passed unconditionally and the
    kernel branches on a flag, because numba does not accept a parameter
    whose rank varies per call.

All `rng.random` and `rng.uniform` draws stay in Python in the same
order, with the same argument shapes and the same conditional skips on
empty selections — so the PCG64 sequence is preserved exactly.

Bit-identical to vanilla on `pmt_afterpulses`: every per-row float op
(scalar division, halving, comparison, argmin) follows the same IEEE-754
sequence as the equivalent numpy expression.

Result on K40-class workloads: ~1.4-1.7x reduction in the PMTAfterPulses
plugin's wall time across ER backgrounds and Pb212.

* ci(pre-commit): pin docformatter to python3.12

pre-commit.ci runs hook environments under python3.14 by default.
docformatter v1.7.7's transitive dependency `untokenize 0.1.1` uses
`ast.Constant.s`, an attribute removed in python3.14, so the install
fails when pre-... (continued)

Coverage Stats

47 of 100 new or added lines in 1 file covered. (47.0%)

2 existing lines in 1 file now uncovered.

2759 of 3773 relevant lines covered (73.12%)

2.91 hits per line