Skip to content

Commit

Permalink
Use max workspace for cudnn conv algo search
Browse files Browse the repository at this point in the history
  • Loading branch information
WolframRhodium authored Dec 5, 2021
1 parent e613019 commit 4fa4365
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ struct CUDAExecutionProviderInfo {
CUDAExecutionProviderExternalAllocatorInfo external_allocator_info{};
// By default use fix workspace size (32M) for Conv algo search, the final algo might not be the best.
// If set to true, try to use as much as possible memory for algo search.
bool cudnn_conv_use_max_workspace{false};
bool cudnn_conv_use_max_workspace{true};

static CUDAExecutionProviderInfo FromProviderOptions(const ProviderOptions& options);
static ProviderOptions ToProviderOptions(const CUDAExecutionProviderInfo& info);
Expand Down

0 comments on commit 4fa4365

Please sign in to comment.