Fix missing DetectRocm() in FindSolutionsImpl() where options = nullp…

…tr (#2288)
ROCm · Aug 2, 2023 · a84ad36 · a84ad36
1 parent 312bc9d
commit a84ad36
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/problem.cpp b/src/problem.cpp
@@ -258,7 +258,9 @@ std::vector<Solution> Problem::FindSolutionsImpl(Handle& handle,
     }
     else
     {
-        const auto workspace_max = conv_desc.GetWorkSpaceSize({&handle}, conv_problem);
+        auto tmp_ctx = ExecutionContext{&handle};
+        tmp_ctx.DetectRocm();
+        const auto workspace_max = conv_desc.GetWorkSpaceSize(tmp_ctx, conv_problem);
         workspace_size           = std::min(options.workspace_limit, workspace_max);
         owned_workspace          = workspace_size != 0 ? handle.Create(workspace_size) : nullptr;
         workspace                = owned_workspace.get();