Skip to content

Add more skip nvMolKit flags to benchmarks#189

Merged
scal444 merged 2 commits into
NVIDIA-BioNeMo:mainfrom
scal444:split/skip-flags
Jun 1, 2026
Merged

Add more skip nvMolKit flags to benchmarks#189
scal444 merged 2 commits into
NVIDIA-BioNeMo:mainfrom
scal444:split/skip-flags

Conversation

@scal444
Copy link
Copy Markdown
Collaborator

@scal444 scal444 commented Jun 1, 2026

The similarity one is annoying because it uses pyperf, which actually subprocesses the entire file for multiple runs.

scal444 added 2 commits May 29, 2026 14:12
…ity)

tfd_bench: `--skip-nvmolkit` to mirror the pre-existing `--skip-rdkit`,
allowing RDKit-only runs (faster wall time on large sweeps).

cross_similarity_bench: `--no-rdkit`/`--no-nvmolkit` plumbed via direct
sys.argv inspection at import time, since pyperf's child worker processes
re-import this module and re-parse argv (registered argparse flags do not
propagate consistently to them).
@scal444 scal444 requested a review from evasnow1992 June 1, 2026 13:21
Copy link
Copy Markdown
Collaborator

@evasnow1992 evasnow1992 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for adding these flags. The changes look good to me.
We may consider consolidating similar flag names across the benchmark scripts for consistency as a future follow-up.

@scal444 scal444 merged commit 89e0a7c into NVIDIA-BioNeMo:main Jun 1, 2026
8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants