Can I compile a modified version of tensorpipe and plug into Pytorch?

pytorch / tensorpipe

A tensor-aware point-to-point communication primitive for machine learning

Other

248 stars 77 forks source link

Can I compile a modified version of tensorpipe and plug into Pytorch? #438

Closed eedalong closed 2 years ago

lw commented 2 years ago

Unfortunately no, TensorPipe is linked statically into PyTorch, so your only option is to rebuild PyTorch from source after modifying the TensorPipe submodule within PyTorch to point to whatever commit you want.