auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-16 13:25:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use heuristic search for CUDA

Browse Source
This commit is contained in:
Michael Hansen
2023-11-22 20:40:14 -06:00
parent 6c5e283439
commit baad6639ed
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/cpp/piper.cpp
View File

@@ -268,6 +268,7 @@ void loadModel(std::string modelPath, ModelSession &session, bool useCuda) {
if (useCuda) {
// Use CUDA provider
OrtCUDAProviderOptions cuda_options{};
cuda_options.cudnn_conv_algo_search = OrtCudnnConvAlgoSearchHeuristic;
session.options.AppendExecutionProvider_CUDA(cuda_options);
}