search

facebookresearch / moolib

A library for distributed ML training with PyTorch
MIT License
366 stars 20 forks source link

Add missing headers to fix build on some platforms #43

Closed xiaomengy closed 2 years ago

xiaomengy commented 2 years ago

There is a missing #include in tensor.h which leads to some build failures on some machines. This PR adds it back.