Passsing device info to requiredWorkspaceSize()

ROCm · Sep 11, 2024 · ab1c6f3 · ab1c6f3
1 parent 9f5ac5d
commit ab1c6f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/library/src/hcc_detail/rocsparselt/src/tensile_host.cpp b/library/src/hcc_detail/rocsparselt/src/tensile_host.cpp
@@ -967,7 +967,7 @@ rocsparselt_status getBestSolutions(const RocsparseltContractionProblem<Ti, To,
     {
         auto solution                  = solutions[i];
         configs[i].index               = solution->index;
-        configs[i].max_workspace_bytes = solution->requiredWorkspaceSize(tensile_prob);
+        configs[i].max_workspace_bytes = solution->requiredWorkspaceSize(tensile_prob, *hardware);
         configs[i].use_bias            = tensile_prob.useBias();
         configs[i].use_scale_alpha_vec = tensile_prob.useScaleAlphaVec();
     }