Skip to content

Commit

Permalink
modify cudnn strategy
Browse files Browse the repository at this point in the history
  • Loading branch information
unsky committed Sep 25, 2018
1 parent ec70591 commit 64e05f2
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/sita/dlflow/operators/convolution.cu
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,6 @@ void Convolution<Dtype>::forward(){

// Synchronize the work across groups, each of which went into its own
// stream, by launching an empty kernel into the default (null) stream.
// NOLINT_NEXT_LINE(whitespace/operators)
sync_conv_groups<<<1, 1>>>();
}
};
Expand Down Expand Up @@ -114,7 +113,6 @@ void Convolution<Dtype>::backward(){
}
// Synchronize the work across groups, each of which went into its own
// stream, by launching an empty kernel into the default (null) stream.
// NOLINT_NEXT_LINE(whitespace/operators)
sync_conv_groups<<<1, 1>>>();
}
};
Expand Down

0 comments on commit 64e05f2

Please sign in to comment.