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

[Dependency Update] Upgrade CI to use latest cuDNN #14950

Merged
merged 9 commits into from
May 17, 2019
Merged

[Dependency Update] Upgrade CI to use latest cuDNN #14950

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
5bfc7ff
bump up the cuDNN version
stu1130 May 8, 2019
159eb70
fall back to fp32
stu1130 May 14, 2019
713551f
add fallback data type
stu1130 May 14, 2019
fd4a7d0
remove the duplicate one
stu1130 May 14, 2019
874e8ab
reenable the memory check
stu1130 May 14, 2019
95d8efc
add comment
stu1130 May 15, 2019
b8a2a1c
fix the linter
stu1130 May 15, 2019
609f014
merge the conflict
stu1130 May 16, 2019
9cc2eb3
retrigger CI
stu1130 May 16, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
reenable the memory check
  • Loading branch information
@stu1130
stu1130 committed May 16, 2019
commit 874e8ab9067ff77e9b561527c512e8a7e5cee170
2 changes: 1 addition & 1 deletion src/operator/rnn-inl.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1398,7 +1398,7 @@ class RNNOp {
x_desc_vec_[0],
&cudnn_param_size,
dtype_));
// CHECK_EQ(w.shape_[0] * sizeof(DType), cudnn_param_size);
CHECK_EQ(w.shape_[0] * sizeof(DType), cudnn_param_size);
// Set param descriptors
int dim_w[3] = {1, 1, 1};
dim_w[0] = w.shape_[0];
Expand Down