From 5ae7d2437ddef370bd0dd7747e84bb6841543445 Mon Sep 17 00:00:00 2001
From: Jeff Daily <jeff.daily@amd.com>
Date: Mon, 14 Oct 2024 17:55:53 +0000
Subject: [PATCH] fix compute type for integer_gemm.cc

---
 onnxruntime/core/providers/rocm/integer_gemm.cc | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/onnxruntime/core/providers/rocm/integer_gemm.cc b/onnxruntime/core/providers/rocm/integer_gemm.cc
index 2be2b753cd573..2d6ee89239cee 100644
--- a/onnxruntime/core/providers/rocm/integer_gemm.cc
+++ b/onnxruntime/core/providers/rocm/integer_gemm.cc
@@ -1,8 +1,6 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
-#include <hip/hip_runtime.h>
-#include <hipblas/hipblas.h>
 #include "core/providers/rocm/shared_inc/integer_gemm.h"
 
 #include "core/common/safeint.h"
@@ -60,7 +58,7 @@ Status GemmInt8(int m, int n, int k,
       lda_aligned == lda ? a : a_padded.get(), HIP_R_8I, lda_aligned,
       &beta,
       c, HIP_R_32I, ldc,
-      HIPBLAS_COMPUTE_32F,
+      HIPBLAS_COMPUTE_32I,
       HIPBLAS_GEMM_DEFAULT));
   return Status::OK();
 }