Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Raise RuntimeError when using cudnn_fast without cudnn #8499

Merged
merged 2 commits into from
Dec 4, 2019

Conversation

emcastillo
Copy link
Member

@emcastillo emcastillo commented Dec 4, 2019

Convolution2d should not use cudnn_fast compute mode if cudnn is not enabled.

The changes required for this involve modifying im2col and col2im to support NHWC layout

The compute mode can only be checked at the link creation time but the associated function checks use_cudnn at forward time. Ideally we should check this at creation time in the link, but since chainer allows to dynamically change some of the backend properties based on the input I deferred the check to the function forward.

@emcastillo
Copy link
Member Author

Extracted them to another class

@emcastillo emcastillo force-pushed the fix_nhwc_conv branch 2 times, most recently from a5b1bc6 to ff4b410 Compare December 4, 2019 10:01
@emcastillo
Copy link
Member Author

PTAL

@niboshi
Copy link
Member

niboshi commented Dec 4, 2019

Thanks
Jenkins, test this please

@niboshi niboshi self-assigned this Dec 4, 2019
@niboshi niboshi added cat:enhancement Implementation that does not break interfaces. st:test-and-merge State indicating that pull request is approved by a reviewer and can be merged after CI passes. labels Dec 4, 2019
@niboshi niboshi added this to the v7.0.0 milestone Dec 4, 2019
@chainer-ci
Copy link
Member

Jenkins CI test (for commit ff02372, target branch master) succeeded!

@niboshi niboshi removed the st:test-and-merge State indicating that pull request is approved by a reviewer and can be merged after CI passes. label Dec 4, 2019
@kmaehashi
Copy link
Member

pfnCI, test this please.

@chainer-ci
Copy link
Member

Jenkins CI test (for commit e224c04, target branch master) succeeded!

@niboshi
Copy link
Member

niboshi commented Dec 4, 2019

LGTM

@niboshi niboshi merged commit d5909f7 into chainer:master Dec 4, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cat:enhancement Implementation that does not break interfaces.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants