saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-25 22:26:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[ROCm] change miopen_conv_use_max_workspace=true (#14982)

Browse source 
Change miopen_conv_use_max_workspace=true to get best algorithm during
`miopenFindConvolutionForwardAlgorithm` process.
This commit is contained in:
PeixuanZuo 2023-03-13 16:19:23 +08:00 committed by GitHub
parent 448e989df8
commit c55f347689
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/core/providers/rocm/rocm_execution_provider_info.h
View file

@ -56,7 +56,10 @@ struct ROCMExecutionProviderInfo {
// arena config.
OrtArenaCfg* default_memory_arena_cfg{nullptr};
ROCMExecutionProviderExternalAllocatorInfo external_allocator_info{};
bool miopen_conv_use_max_workspace{false};
// By default, try to use as much as possible memory for algo search.
// If set to false, use fix workspace size (32M) for Conv algo search, the final algo might not be the best.
bool miopen_conv_use_max_workspace{true};
rocm::TunableOpInfo tunable_op{};