From c33391329a7cb85b2ac15257e0355d3f175cfbdb Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Wed, 20 Oct 2021 10:37:58 -0700 Subject: [PATCH] Add QuantizeLinear and DequantizeLinear to ROCm EP (#9401) --- .../core/providers/rocm/rocm_execution_provider.cc | 8 ++++---- tools/ci_build/amd_hipify.py | 4 ---- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/onnxruntime/core/providers/rocm/rocm_execution_provider.cc b/onnxruntime/core/providers/rocm/rocm_execution_provider.cc index 3b35f11dde..fef51b55cd 100644 --- a/onnxruntime/core/providers/rocm/rocm_execution_provider.cc +++ b/onnxruntime/core/providers/rocm/rocm_execution_provider.cc @@ -1583,10 +1583,10 @@ static Status RegisterRocmKernels(KernelRegistry& kernel_registry) { // BuildKernelCreateInfo, // BuildKernelCreateInfo, // BuildKernelCreateInfo, - // BuildKernelCreateInfo, - // BuildKernelCreateInfo, - // BuildKernelCreateInfo, - // BuildKernelCreateInfo, + BuildKernelCreateInfo, + BuildKernelCreateInfo, + BuildKernelCreateInfo, + BuildKernelCreateInfo, // opset 11 BuildKernelCreateInfo, diff --git a/tools/ci_build/amd_hipify.py b/tools/ci_build/amd_hipify.py index 49ae16cc68..3c6afb0b78 100644 --- a/tools/ci_build/amd_hipify.py +++ b/tools/ci_build/amd_hipify.py @@ -133,10 +133,6 @@ provider_excluded_files = [ 'shared_inc/cuda_call.h', 'shared_inc/fpgeneric.h', 'shared_inc/integer_gemm.h', - 'tensor/quantize_linear.cc', - 'tensor/quantize_linear.cu', - 'tensor/quantize_linear.cuh', - 'tensor/quantize_linear.h', 'tensor/resize.cc', 'tensor/resize.h', 'tensor/resize_impl.cu',