saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-03 03:58:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added miopenGetConvolutionSpatialDim if ROCm5.5 (#14772)

Browse source 
The API should be `miopenGetConvolutionSpatialDim(cdesc, &spatial_dim)`, NOT `miopenGetConvolutionDescriptorSize(cdesc, &spatial_dim)`
This commit is contained in:
Zachary Streeter 2023-03-02 18:02:32 -06:00 committed by GitHub
parent 0ebe8e34f8
commit 6e2ca15140
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/contrib_ops/rocm/fused_conv.cc
View file

@ -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.