mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #11526 from atgctg:patch-2

Browse Source 
PiperOrigin-RevId: 461903614
This commit is contained in:
jax authors 2022-07-19 09:42:30 -07:00
commit 97fe2262d3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/faq.rst
View File

@ -582,7 +582,7 @@ Buffer donation
(This feature is implemented only for TPU and GPU.)
When JAX executes a computation it uses buffers on the device for all inputs and outputs.
If you know than one of the inputs is not needed after the computation, and if it
If you know that one of the inputs is not needed after the computation, and if it
matches the shape and element type of one of the outputs, you can specify that you
want the corresponding input buffer to be donated to hold an output. This will reduce
the memory required for the execution by the size of the donated buffer.
@ -784,4 +784,4 @@ See :class:`jax.errors.ConcretizationTypeError`
.. _Heaviside Step Function: https://en.wikipedia.org/wiki/Heaviside_step_function
.. _Sigmoid Function: https://en.wikipedia.org/wiki/Sigmoid_function
.. _Sigmoid Function: https://en.wikipedia.org/wiki/Sigmoid_function