Replace libtorch with dlpack

hpcaitech / TensorNVMe

A Python library transfers PyTorch tensors between CPU and NVMe

93 stars 18 forks source link

Closed oahzxl closed 2 years ago

oahzxl commented 2 years ago

add a test for offload first

oahzxl commented 2 years ago

polish cmake and use dlpack remain to be done