From 8495918c0fa61c42637a69f716863cbaeb442dc4 Mon Sep 17 00:00:00 2001 From: stsokolo Date: Wed, 2 Oct 2024 15:17:19 +0200 Subject: [PATCH] Add rocm to transformers/benchmark.py script (#68) Co-authored-by: Stefan Sokolovic --- 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 92cb1b4369acf..59204aae7add9 100644 --- a/onnxruntime/python/tools/transformers/benchmark.py +++ b/onnxruntime/python/tools/transformers/benchmark.py @@ -788,7 +788,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