/pytorch-rle

A implement of run-length encoding for Pytorch tensor using CUDA

Primary LanguageCuda

Pytorch Run-Length Encoding

this repo contains two implmentations:

  • RLE-raw-cuda: It's written using pure cuda lib
  • RLE-thrust: It's written using a higher level lib thrust