NVIDIA-Merlin / core

Core Utilities for NVIDIA Merlin
Apache License 2.0
19 stars 14 forks source link

Add `cpu()` and `gpu()` transfer methods to `TensorColumn/Table` #232

Closed karlhigley closed 1 year ago

karlhigley commented 1 year ago

These methods provide a way to move explicitly and intentionally move data across devices, which is otherwise not supported by the existing cross-framework data transfer methods. (We avoided integrating cross-device transfer there in order to avoid hidden performance costs.)