Skip to content

Commit

Permalink
Fix only use ext api when use_ext is enabled in the matmul test.
Browse files Browse the repository at this point in the history
  • Loading branch information
vin-huang committed Jul 10, 2023
1 parent bbc7460 commit 0043cae
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion clients/include/testing_matmul.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -1427,10 +1427,11 @@ void testing_matmul(const Arguments& arg)

if(!do_grouped_gemm)
{
CHECK_HIPBLASLT_ERROR(gemm.initialize(heuristicResult[0].algo, *dWorkspace, stream));

if(arg.use_ext)
{
CHECK_HIPBLASLT_ERROR(
gemm.initialize(heuristicResult[0].algo, *dWorkspace, stream));
for(int i = 0; i < number_cold_calls; i++)
CHECK_HIPBLASLT_ERROR(gemm.run(stream));
CHECK_HIP_ERROR(hipStreamSynchronize(stream));
Expand Down

0 comments on commit 0043cae

Please sign in to comment.