mk-61 opened a new pull request #19318: URL: https://github.com/apache/incubator-mxnet/pull/19318
## Description ## The patch fixes possible CUDA Invalid Memory Access when fusing amp_multicast op with inputs of different sizes. It has been observed on Yolo3 model from GluonCV. ## Checklist ## ### Essentials ### - [X] Changes are complete (i.e. I finished coding on this PR) @ptrendx ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
