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

Commit

Permalink
merge conflict
Browse files Browse the repository at this point in the history
  • Loading branch information
roywei committed Apr 7, 2020
1 parent 977428b commit b8efb3d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 6 deletions.
5 changes: 2 additions & 3 deletions include/mxnet/resource.h
Original file line number Diff line number Diff line change
Expand Up @@ -202,9 +202,8 @@ struct Resource {
void get_cudnn_dropout_desc(
cudnnDropoutDescriptor_t *dropout_desc,
mshadow::Stream<gpu> *stream,
const float dropout, uint64_t seed,
const std::string &name = MXNET_RESOURCE_DEFAULT_NAME_FARG("cudnn_dropout_state"),
bool reset) const;
const float dropout, uint64_t seed, bool reset,
const std::string &name = MXNET_RESOURCE_DEFAULT_NAME_FARG("cudnn_dropout_state")) const;
#endif // MXNET_USE_CUDNN == 1

/*!
Expand Down
5 changes: 2 additions & 3 deletions src/resource.cc
Original file line number Diff line number Diff line change
Expand Up @@ -428,9 +428,8 @@ void* Resource::get_host_space_internal(size_t size) const {
void Resource::get_cudnn_dropout_desc(
cudnnDropoutDescriptor_t *dropout_desc,
mshadow::Stream<gpu> *stream,
const float dropout, uint64_t seed,
const std::string &name,
bool reset) const {
const float dropout, uint64_t seed, bool reset,
const std::string &name) const {

CHECK_EQ(req.type, ResourceRequest::kCuDNNDropoutDesc);
auto state_space = static_cast<resource::SpaceAllocator*>(ptr_);
Expand Down

0 comments on commit b8efb3d

Please sign in to comment.