Skip to content

Comments

Add jax.numpy for testing#2488

Open
havogt wants to merge 1 commit intoGridTools:mainfrom
havogt:add_jnp_support
Open

Add jax.numpy for testing#2488
havogt wants to merge 1 commit intoGridTools:mainfrom
havogt:add_jnp_support

Conversation

@havogt
Copy link
Contributor

@havogt havogt commented Feb 20, 2026

requires #2442

Comment on lines 63 to 68
numpy_execution = EmbeddedDummyBackend(
"EmbeddedNumPy", next_allocators.StandardCPUFieldBufferAllocator()
)
cupy_execution = EmbeddedDummyBackend(
"EmbeddedCuPy", next_allocators.StandardGPUFieldBufferAllocator()
)
Copy link
Contributor Author

Choose a reason for hiding this comment

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

Suggested change
)
numpy_execution = EmbeddedDummyBackend(
"EmbeddedNumPy", np
)
cupy_execution = EmbeddedDummyBackend(
"EmbeddedCuPy", cp
)

cp from some import that is possibly None

Copy link
Contributor Author

Choose a reason for hiding this comment

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

or import here...

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.

1 participant