Skip to content

Commit 2e0b71e

Browse files
committed
fix: gpu benhcmarks actually use gpu
1 parent c4557b3 commit 2e0b71e

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

benchmarking/bench_scripts/bench.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -318,7 +318,8 @@ def main(
318318
score_sizes: list[ScoreSize],
319319
parasail_instruction_sets: list[ParasailInstructionSet] = [],
320320
output_dir: Path,
321-
devices: int = 1
321+
devices: int = 1,
322+
iterations: int = 3
322323
):
323324

324325
# score_sizes = ["byte", "word", "adaptive"]
@@ -342,7 +343,7 @@ def main(
342343
score_size=score_size,
343344
scoring_matrix="Blosum62",
344345
output_file=str(output_dir / "ish-aligner-result.csv"),
345-
iterations=3,
346+
iterations=iterations,
346347
algo=ish_algorithm,
347348
devices=device + 1,
348349
)

pixi.toml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,11 +168,12 @@ bash -c '
168168
fi &&
169169
python benchmarking/bench_scripts/bench.py \
170170
--ish-binaries $(readlink -f ./ish-aligner-gpu) \
171-
--ish-algorithm striped-semi-global \
171+
--ish-algorithm basic-semi-global-gpu-parallel \
172172
--score-size word \
173173
--query-seqs-base-dir ./bench/data/parasail/data/ \
174-
--ref-data bench/data/refdata/uniprot_sprot_10x.fasta \
174+
--ref-data bench/data/refdata/uniprot_sprot_5x.fasta \
175175
--devices 4 \
176+
--iterations 1 \
176177
--output-dir ./bench_results/gpu/ \
177178
> ./bench_results/gpu/benchmark_results.csv &&
178179
python benchmarking/bench_scripts/plot.py \

0 commit comments

Comments
 (0)