diff options
author | 2018-07-26 23:07:53 -0700 | |
---|---|---|
committer | 2018-07-26 23:10:45 -0700 | |
commit | 21fbc78e69b3e79f883161601f1ccc62be42a06e (patch) | |
tree | 9438003d582541359062700cbb9e7f0e95a7a894 /third_party/backports_weakref.BUILD | |
parent | 19defc68c99483049e14ec18718cb037f1ca050b (diff) |
Set the correct context when calling cudnnCreate.
When running with multiple devices, using the wrong context will lead to
a check-fail when trying to set a stream that has been created with a different
context.
This resolves a check-fail on resnet50 with 8 GPUs.
PiperOrigin-RevId: 206274741
Diffstat (limited to 'third_party/backports_weakref.BUILD')
0 files changed, 0 insertions, 0 deletions