Skip to content
This repository was archived by the owner on Jan 26, 2026. It is now read-only.

Better initialization for wave and shallow water benchmarks#19

Merged
tkarna merged 4 commits intomainfrom
bye-bye-fromfunction
Nov 11, 2024
Merged

Better initialization for wave and shallow water benchmarks#19
tkarna merged 4 commits intomainfrom
bye-bye-fromfunction

Conversation

@tkarna
Copy link
Collaborator

@tkarna tkarna commented Nov 7, 2024

  • Replace fromfunction in initialization by reshape and permute_dims
  • Exclude JIT compilation time from timing. This is consistent with other benchmarks.

@tkarna tkarna force-pushed the bye-bye-fromfunction branch from 32d491c to aba4ce8 Compare November 7, 2024 21:18
@tkarna tkarna requested a review from fschlimb November 7, 2024 21:29
Copy link
Contributor

@fschlimb fschlimb left a comment

Choose a reason for hiding this comment

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

Yey!

@tkarna tkarna merged commit 9506c28 into main Nov 11, 2024
@tkarna tkarna deleted the bye-bye-fromfunction branch November 11, 2024 05:43
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants