Skip to content
Snippets Groups Projects
Commit 1557287c authored by Son Tuan Vu's avatar Son Tuan Vu Committed by TensorFlower Gardener
Browse files

[xla:gpu] Handle arbitrary number of arguments passed to AddressComputationThunk 

Passing the correct buffers to embedded thunk when AddressComputationThunk itself is getting an arbitrary number of buffers.

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/63954 from elfringham:fix_312_setuptools da09611b
PiperOrigin-RevId: 617185292
parent 558d9806
Hide whitespace changes
Inline Side-by-side
Showing
with 262 additions and 94 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment