diff --git a/src/registered_memory.cc b/src/registered_memory.cc index 3a56e357..48712e5c 100644 --- a/src/registered_memory.cc +++ b/src/registered_memory.cc @@ -14,10 +14,8 @@ namespace mscclpp { -RegisteredMemory::Impl::Impl(void* data, size_t size, int rank, TransportFlags transports, - Communicator::Impl& commImpl) { - Impl(data, size, size, rank, transports, commImpl); -} +RegisteredMemory::Impl::Impl(void* data, size_t size, int rank, TransportFlags transports, Communicator::Impl& commImpl) + : Impl(data, size, size, rank, transports, commImpl) {} RegisteredMemory::Impl::Impl(void* data, size_t size, size_t pitch, int rank, TransportFlags transports, Communicator::Impl& commImpl) @@ -80,6 +78,7 @@ MSCCLPP_API_CPP TransportFlags RegisteredMemory::transports() { return pimpl->tr MSCCLPP_API_CPP std::vector RegisteredMemory::serialize() { std::vector result; std::copy_n(reinterpret_cast(&pimpl->size), sizeof(pimpl->size), std::back_inserter(result)); + std::copy_n(reinterpret_cast(&pimpl->pitch), sizeof(pimpl->pitch), std::back_inserter(result)); std::copy_n(reinterpret_cast(&pimpl->rank), sizeof(pimpl->rank), std::back_inserter(result)); std::copy_n(reinterpret_cast(&pimpl->hostHash), sizeof(pimpl->hostHash), std::back_inserter(result)); std::copy_n(reinterpret_cast(&pimpl->transports), sizeof(pimpl->transports), std::back_inserter(result));