pytorch
aff693ab - Ensure MIOpen is called on same stream as operator for RNN (#30672)

Commit
6 years ago
Ensure MIOpen is called on same stream as operator for RNN (#30672) Summary: To ensure synchronization between copying of weights in RNN wei buf, and the operation, both the pyTorch operator as well as underlying MIOpen call must be on the same HIP stream. This is also consistent with MIOpen calls in other pyTorch operators ezyang iotamudelta Pull Request resolved: https://github.com/pytorch/pytorch/pull/30672 Differential Revision: D18785683 Pulled By: bddppq fbshipit-source-id: 144611046cb70cfe450680295734203f253ac6e2
Author
ashish
Parents
Loading