Skip to content

[FEATURE] Add performance mode #1330

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
hughperkins merged 8 commits into from
Jun 30, 2025
Merged

[FEATURE] Add performance mode #1330

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
41349c3
add perofrmance mode
hughperkins Jun 28, 2025
5a64e19
change how performance mode is applied
hughperkins Jun 28, 2025
0a839c7
move back into args
hughperkins Jun 28, 2025
eb8e629
Merge branch 'main' into hp/add-performance-mode
duburcqa Jun 30, 2025
1aec7a9
Merge branch 'main' into hp/add-performance-mode
YilingQiao Jun 30, 2025
8c6574c
add info about perofrmance mode
hughperkins Jun 30, 2025
af69fac
Merge branch 'hp/add-performance-mode' of github.com:hughperkins/Gene…
hughperkins Jun 30, 2025
d9c30be
update comment
hughperkins Jun 30, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions genesis/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ def init(
backend=None,
theme="dark",
logger_verbose_time=False,
performance_mode: bool = False, # True: compilation up to 6x slower (GJK), but runs ~1-5% faster
):
# Consider Genesis as initialized right away
global _initialized
Expand Down Expand Up @@ -172,6 +173,12 @@ def init(
torch.backends.cudnn.benchmark = False
logger.info("Beware running Genesis in debug mode dramatically reduces runtime speed.")

if not performance_mode:
logger.info(
"Consider setting 'performance_mode=True' in production to maximise runtime speed, if significantly "
"increasing compilation time is not a concern."
)

if seed is not None:
global SEED
SEED = seed
Expand All @@ -197,6 +204,7 @@ def init(
force_scalarize_matrix=True,
# Turning off 'advanced_optimization' is causing issues on MacOS
advanced_optimization=True,
cfg_optimization=performance_mode,
fast_math=not debug,
default_ip=ti_int,
default_fp=ti_float,
Expand Down