Closed copybara-service[bot] closed 1 week ago
Remove PjRtStreamExecutorMemorySpace specialization from PjRtStreamExecutorClient
PjRtStreamExecutorMemorySpace
PjRtStreamExecutorClient
This allows us to use existing host memory spaces for GPUs without having to reimplement them for stream executor clients.
Remove
PjRtStreamExecutorMemorySpacespecialization fromPjRtStreamExecutorClientThis allows us to use existing host memory spaces for GPUs without having to reimplement them for stream executor clients.