xla
Transfer data directly to the device
#5752
Merged
Overview
Commits
12
Changes
View On GitHub

Transfer data directly to the device #5752

will-cromar merged 12 commits into master from wcromar/remove-populate-fn
will-cromar
will-cromar Remove `populate_fn` from `TensorSource`
d2dfe247
will-cromar will-cromar added runtime
will-cromar Make TensorSource an interface
fad33dd6
will-cromar Re-enable pjrt_computation_client_test
7b9cfe39
will-cromar server -> device
63845167
will-cromar add comment
8d9f2d1e
will-cromar fix outbound data metric
c0cca42d
will-cromar formatting
ff2ffb94
will-cromar implement byte_strides in TensorSource
ab88d502
will-cromar will-cromar changed the title Remove `populate_fn` from `TensorSource` Transfer data directly to the device 2 years ago
will-cromar more formatting
904eae51
will-cromar remove extra deps
59e73ef1
will-cromar add missing deps
9b7d481d
will-cromar Revert "server -> device"
2a6433dc
will-cromar
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
will-cromar will-cromar marked this pull request as ready for review 2 years ago
JackCaoG
JackCaoG approved these changes on 2023-11-01
will-cromar will-cromar merged 5ca36cf5 into master 2 years ago
yeounoh

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
runtime
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum