Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[GPU] Fix test_gluon_gpu:test_lstmp #14529

Merged
merged 1 commit into from
Mar 27, 2019

Conversation

perdasilva
Copy link
Contributor

@perdasilva perdasilva commented Mar 26, 2019

Description

The way CI is at the moment, some failing GPU tests are going undetected. I'm putting together a PR (#14513) to update CI do CUDA v10 and fix a bug in the way the cuDNN version gets used in the tests.

In this PR, I fix one such failing test introduced in PR #14219, where the function signature for check_rnn_layer_forward was changed and a context parameter added with a default value of mx.cpu().

The test_lstmp test fails because it uses the check_rnn_layer_forward function without setting the context to mx.gpu(0).

In this PR, I remove the default value to force the user to set the context. Thus avoiding such errors in the future.

For an example failure, see: http://jenkins.mxnet-ci.amazon-ml.com/blue/organizations/jenkins/mxnet-validation%2Funix-gpu/detail/PR-14513/5/pipeline

@lebeg
Copy link
Contributor

lebeg commented Mar 26, 2019

lgtm

@lebeg
Copy link
Contributor

lebeg commented Mar 26, 2019

It would be great if you could mention test_lstmp in the title or description explicitly to ease the search.

@perdasilva perdasilva changed the title Fix lstmp GPU test [GPU] Fix test_gluon_gpu:test_lstmp Mar 26, 2019
@abhinavs95
Copy link
Contributor

@mxnet-label-bot update [CI, Test]

Copy link
Member

@lanking520 lanking520 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@lanking520 lanking520 merged commit 67c10f9 into apache:master Mar 27, 2019
@perdasilva
Copy link
Contributor Author

@lanking520 thank you!

@perdasilva perdasilva deleted the fix_check_rnn_layer_fwd branch March 27, 2019 07:36
vdantu pushed a commit to vdantu/incubator-mxnet that referenced this pull request Mar 31, 2019
ZhennanQin pushed a commit to ZhennanQin/incubator-mxnet that referenced this pull request Apr 3, 2019
haohuanw pushed a commit to haohuanw/incubator-mxnet that referenced this pull request Jun 23, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants