[GitHub] wkcn edited a comment on issue #14278: use cudnn for dropout by default

[GitBox <gi...@apache.org>]

wkcn edited a comment on issue #14278: use cudnn for dropout by default
URL: https://github.com/apache/incubator-mxnet/pull/14278#issuecomment-468178186
 
 
   Thanks for your contribution!
   
   unix-gpu CI raises the following exception:
   ```bash
   C++ exception with description "[07:12:07] /work/mxnet/src/resource.cc:429: Check failed: state_space->ctx.dev_id == stream->dev_id (0 vs. -1) The device id of cudnn dropout state space doesn't match that from stream.
   ```
   
   Maybe we should modify the code of `tests/cpp/include/test_op.h`.
   `opContext_.run_ctx.stream = mshadow::NewStream<gpu>(true, true);` -> `opContext_.run_ctx.stream = mshadow::NewStream<gpu>(true, true, 0);`

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services