jax
562e9e8d - Fix an incorrect output for jnp.cumsum.

Commit

1 year ago

Fix an incorrect output for jnp.cumsum. If dtype=bool but a non-bool input is passed, we should test for non-equality with zero rather than performing a cast to integer.

References

#23873 - Fix an incorrect output for jnp.cumsum.

Author

hawkinsp

Committer

hawkinsp

Parents

80cb821a

jax 562e9e8d - Fix an incorrect output for jnp.cumsum.

jax
562e9e8d - Fix an incorrect output for jnp.cumsum.