DeepLearningExamples icon indicating copy to clipboard operation
DeepLearningExamples copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA

[PyTorch/DLRM] __restrict issue

Open wchen61 opened this issue 2 years ago • 0 comments

Is this code a typo and should be half *__restrict grad?

https://github.com/NVIDIA/DeepLearningExamples/blob/b1fc3c46f508201405c76e4811c036b05e9773d7/PyTorch/Recommendation/DLRM/dlrm/cuda_src/dot_based_interact/dot_based_interact_tf32_bwd.cu#L32

wchen61 avatar Nov 10 '22 03:11 wchen61