torcharrow icon indicating copy to clipboard operation
torcharrow copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Any plan to support GPU?

Open whbldhwj opened this issue 3 years ago • 1 comments
trafficstars

Hi,

This looks like a really interesting project! I saw currently torcharrow runs on CPU with the Velox backend. Just wondering any plan to offload some of the ops to GPU? If exists, what's the technical path? Any plan to reuse the cudf/rapids library?

Thanks, Jie

whbldhwj avatar Sep 20 '22 23:09 whbldhwj

Thanks @whbldhwj ! we do plan on supporting GPU. Current prototype is based on PyTorch itself. Would you mind sharing if have particular use cases that can benefit from GPU processing?

dracifer avatar Sep 28 '22 00:09 dracifer