HugeCTR icon indicating copy to clipboard operation
HugeCTR copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-Merlin

Fix wrong shape of d_row_ptrs_ for multi emb-tables in <inference_wrapper.hpp>

Open xmh645214784 opened this issue 3 years ago • 2 comments

xmh645214784 avatar Jul 13 '21 02:07 xmh645214784

@zehuanw Please review it. This bug affects the inference of multi-embedding tables with memory leak.

xmh645214784 avatar Jul 19 '21 02:07 xmh645214784

Thank you for the contribution @xmh645214784 ! Could you please check if this issue has been fixed in this commit (on master)? https://github.com/NVIDIA/HugeCTR/commit/dfe5e7ac2238847920c26ea57bac46a017fe55d2

zehuanw avatar Jul 19 '21 03:07 zehuanw