leezu opened a new pull request #19113: URL: https://github.com/apache/incubator-mxnet/pull/19113
Ensure that parameters are retrieved from the correct context when calling the CachedOp, by setting the default context. Prior tests did not catch this issue, as MXNet 2 GPU testing strategy is implemented by changing the default context to gpu context globally: https://github.com/apache/incubator-mxnet/blob/6b01dc24ad89894b1d42021072c3f181b570fbb1/tests/python/gpu/test_deferred_compute_gpu.py#L21-L28 ---------------------------------------------------------------- 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]
