Skip to content

Commit d7efc1d

Browse files
rvangaramdbarnesUCSD
authored andcommitted
Parallelize SPA runs with configurable CPU count
- Parallel SPA calls using min(number of CPUs, samples) - Add `ncpu` parameter to configure cores
1 parent 0cc72cd commit d7efc1d

File tree

4 files changed

+1183
-342
lines changed

4 files changed

+1183
-342
lines changed

SigProfilerAssignment/Analyzer.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ def decompose_fit(
2424
export_probabilities=True,
2525
export_probabilities_per_mutation=False,
2626
volume=None,
27+
ncpu=-1,
2728
):
2829
decomp.spa_analyze(
2930
samples=samples,
@@ -51,6 +52,7 @@ def decompose_fit(
5152
export_probabilities=export_probabilities,
5253
export_probabilities_per_mutation=export_probabilities_per_mutation,
5354
volume=volume,
55+
ncpu=ncpu,
5456
)
5557

5658

@@ -76,6 +78,7 @@ def denovo_fit(
7678
export_probabilities=True,
7779
export_probabilities_per_mutation=False,
7880
volume=None,
81+
ncpu=-1,
7982
):
8083
decomp.spa_analyze(
8184
samples=samples,
@@ -102,6 +105,7 @@ def denovo_fit(
102105
export_probabilities=export_probabilities,
103106
export_probabilities_per_mutation=export_probabilities_per_mutation,
104107
volume=volume,
108+
ncpu=ncpu,
105109
)
106110

107111

@@ -128,6 +132,7 @@ def cosmic_fit(
128132
export_probabilities_per_mutation=False,
129133
sample_reconstruction_plots=False,
130134
volume=None,
135+
ncpu=-1,
131136
):
132137
decomp.spa_analyze(
133138
samples=samples,
@@ -155,4 +160,5 @@ def cosmic_fit(
155160
export_probabilities_per_mutation=export_probabilities_per_mutation,
156161
sample_reconstruction_plots=sample_reconstruction_plots,
157162
volume=volume,
163+
ncpu=ncpu
158164
)

0 commit comments

Comments
 (0)