From efb8703a256a8a7c646ae79974d89baacccac9ca Mon Sep 17 00:00:00 2001
From: stsokolo <stefan.sokolovic2@amd.com>
Date: Tue, 8 Oct 2024 06:57:18 +0200
Subject: [PATCH] [MIGraphX EP Support]Add rocm to transformers/benchmark.py
 script (#22299)

### Description
Add ROCm EP option to benchmark.py script when using int8 quantization.



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
Without this change benchmarks with int8 quantization cannot be run with
ROCm execution provider.
---
 onnxruntime/python/tools/transformers/benchmark.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/onnxruntime/python/tools/transformers/benchmark.py b/onnxruntime/python/tools/transformers/benchmark.py
index 4800c48744..26f8987c76 100644
--- a/onnxruntime/python/tools/transformers/benchmark.py
+++ b/onnxruntime/python/tools/transformers/benchmark.py
@@ -787,7 +787,7 @@ def main():
         logger.error("fp16 is for GPU only")
         return
 
-    if args.precision == Precision.INT8 and args.use_gpu and args.provider != "migraphx":
+    if args.precision == Precision.INT8 and args.use_gpu and args.provider not in ["migraphx", "rocm"]:
         logger.error("int8 is for CPU only")
         return