xla
daada224 - Make from_dlpack handle cuda synchronization implicitly for input tensors that have dlpack and __dlpack_device__ attributes. (#7125)

Commit

1 year ago

Make from_dlpack handle cuda synchronization implicitly for input tensors that have __dlpack__ and __dlpack_device__ attributes. (#7125)

References

Author

vanbasten23

Parents

xla daada224 - Make from_dlpack handle cuda synchronization implicitly for input tensors that have __dlpack__ and __dlpack_device__ attributes. (#7125)