8000 `jax.numpy.put` brings different results with `numpy.put` · Issue #28602 · jax-ml/jax · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

jax.numpy.put brings different results with numpy.put #28602

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

Jump to bottom
Closed
apiqwe opened this issue May 8, 2025 · 1 comment · Fixed by #28610
Closed

jax.numpy.put brings different results with numpy.put #28602

apiqwe opened this issue May 8, 2025 · 1 comment · Fixed by #28610
Assignees
@jakevdp
Labels
bug Something isn't working

Comments

@apiqwe
Copy link
apiqwe commented May 8, 2025

Description

I found that jax.numpy.put brings different results with numpy.put in following case.

import numpy as np
import jax.numpy as jnp

print(np.put(np.array([1]), np.array([0]), np.array([])))
print(jnp.put(jnp.array([1]), jnp.array([0]), jnp.array([])))

Output:

None
[1]

System info (python version, jaxlib version, accelerator, etc.)

jax:    0.6.0
jaxlib: 0.6.0
numpy:  2.2.3
python: 3.10.12 (main, Feb  4 2025, 14:57:36) [GCC 11.4.0]
device info: NVIDIA GeForce RTX 4090-1, 1 local devices"
process_count: 1
platform: uname_result(system='Linux', node='4db45dc420f8', release='6.11.0-25-generic', version='#25~24.04.1-Ubuntu SMP PREEMPT_DYNAMIC Tue Apr 15 17:20:50 UTC 2', machine='x86_64')


$ nvidia-smi
Wed May  7 16:26:08 2025       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.230.02             Driver Version: 535.230.02   CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 4090        Off | 00000000:01:00.0 Off |                  Off |
| 36%   32C    P2              36W / 450W |    407MiB / 24564MiB |      1%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
+---------------------------------------------------------------------------------------+
@apiqwe apiqwe added the bug Something isn't working label May 8, 2025
@jakevdp jakevdp self-assigned this May 8, 2025
@jakevdp jakevdp mentioned this issue May 8, 2025
Merged
@jakevdp
Copy link
Collaborator
jakevdp commented May 8, 2025

Thanks! Should be fixed by #28610

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@jakevdp jakevdp

Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

jnp.put: check inplace before other conditions jakevdp/jax
2 participants
@jakevdp @apiqwe
0