From 6e2ca15140d0bb721dfe0184c070bd1d4c93c775 Mon Sep 17 00:00:00 2001 From: Zachary Streeter <90640993+zstreet87@users.noreply.github.com> Date: Thu, 2 Mar 2023 18:02:32 -0600 Subject: [PATCH] added miopenGetConvolutionSpatialDim if ROCm5.5 (#14772) The API should be `miopenGetConvolutionSpatialDim(cdesc, &spatial_dim)`, NOT `miopenGetConvolutionDescriptorSize(cdesc, &spatial_dim)` --- onnxruntime/contrib_ops/rocm/fused_conv.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/contrib_ops/rocm/fused_conv.cc b/onnxruntime/contrib_ops/rocm/fused_conv.cc index 4049d6dcb1..643dc0359f 100644 --- a/onnxruntime/contrib_ops/rocm/fused_conv.cc +++ b/onnxruntime/contrib_ops/rocm/fused_conv.cc @@ -75,7 +75,7 @@ struct FNVHash { void HashConvolutionDescriptor(miopenConvolutionDescriptor_t cdesc) { int spatial_dim = 1; #if ROCM_VERSION >= 50500 - miopenGetConvolutionDescriptorSize(cdesc, &spatial_dim); + miopenGetConvolutionSpatialDim(cdesc, &spatial_dim); #else // Previous versions of MIOpen doesn't provide API to probe the dimension of a // miopenConvolutionDescriptor_t, so we have to guess.