Groups keyboard shortcuts have been updated
Dismiss
See shortcuts

tf.aliasing vs jax.buffer_donor

18 views
Skip to first unread message

Steeve Morin

unread,
Jan 14, 2025, 4:29:33 PMJan 14
to OpenXLA Discuss
Hi folks.

Is there a reason to use tf.aliasing instead of the higher level jax.buffer_donor ?
It seems that jax.buffer_donor lets the compiler decide which buffer is going to be aliased to the output. I can think of a few example where that could be a problem (swapping for instance?).

What are the recommendations in that case?

Regards,
Reply all
Reply to author
Forward
0 new messages