From ab1c6f30fc48672bde11b7f4e5114c607d077b3f Mon Sep 17 00:00:00 2001
From: Vin Huang <vin.huang@amd.com>
Date: Wed, 11 Sep 2024 14:02:36 +0000
Subject: [PATCH] Passsing device info to requiredWorkspaceSize()

---
 library/src/hcc_detail/rocsparselt/src/tensile_host.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/library/src/hcc_detail/rocsparselt/src/tensile_host.cpp b/library/src/hcc_detail/rocsparselt/src/tensile_host.cpp
index 610936a6..ecd5f738 100644
--- 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();
     }