Commit 21fbc78e authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

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
parent 19defc68
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment