donate_argnums and CPU #6379
Unanswered
davisyoshida
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I wasn't sure if I should open an issue about this or if this is more of an XLA problem but it seems that passing
donate_argnums
tojit
when using CPU doesn't work:I was looking into doing some parameter updates on the CPU and was surprised to find that JITing
jnp.add
didn't allow me to donate the parameter buffer. Is there any way I can force the JIT to write the result to the input buffer to avoid a copy?Beta Was this translation helpful? Give feedback.
All reactions