fix warpctc plugin work with new version warpctc #4530

Merged
merged 1 commit into from Jan 7, 2017

Projects

None yet

2 participants

@yajiedesign
Contributor

No description provided.

@yajiedesign yajiedesign referenced this pull request in baidu-research/warp-ctc Jan 5, 2017
Open

Warp-CTC error on GPU : "cuda memcpy or memset failed" #53

@piiswrong
Member

Are you sure this fix works? In the issue someone said after the fix gpu inference still doesn't work

@yajiedesign
Contributor

i am test it work .
The key point is info.blank_label = 0;`

plugin/warpctc/warpctc-inl.h
@@ -120,7 +120,7 @@ class WarpCTCOp : public Operator {
TBlob data = in_data[warpctc_enum::kData];
TBlob label = in_data[warpctc_enum::kLabel];
CHECK_EQ(data.shape_.ndim(), 2) << "input data shape should be 2 (t*n, p)";
- ctcComputeInfo info; // please build warp-ctc with commit 5bfb46e (cd warp-ctc && git checkout 5bfb46e) NOLINT(*)
+ ctcOptions info;
@piiswrong
piiswrong Jan 5, 2017 Member

cool. add a comment in this line: please updated to latest baidu/warp-ctc

@piiswrong
Member

windows build is failing @yajiedesign

@yajiedesign yajiedesign fix warpctc plugin work with new version warpctc
1baede5
@piiswrong piiswrong merged commit b24c6c5 into dmlc:master Jan 7, 2017

3 checks passed

continuous-integration/appveyor/pr AppVeyor build succeeded
Details
continuous-integration/travis-ci/pr The Travis CI build passed
Details
default Build finished.
Details
@yajiedesign yajiedesign deleted the yajiedesign:master_warpctc branch Jan 12, 2017
@rravu3 rravu3 pushed a commit to rravu3/mxnet that referenced this pull request Jan 21, 2017
@yajiedesign yajiedesign + Rahul Ravu fix warpctc plugin work with new version warpctc (#4530) 833f0aa
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment