Merge branch 'dyn_check_shapes' of github.com:ROCmSoftwarePlatform/AMDMIGraphX into dyn_nms

38582cd5 · charlie · adc77b9f · b6099eeb · 38582cd5
Commit 38582cd5 authored Jul 19, 2022 by charlie
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/include/migraphx/check_shapes.hpp src/include/migraphx/check_shapes.hpp +3 -0

No files found.
--- a/src/include/migraphx/check_shapes.hpp
+++ b/src/include/migraphx/check_shapes.hpp
@@ -186,6 +186,9 @@ struct check_shapes
    {
        if(!this->same([](const shape& s) { return s.max_lens(); }))
            MIGRAPHX_THROW(prefix() + "Dimensions do not match");
+        if(this->any_of([&](const shape& s) { return s.dynamic(); }))
+            if(!this->same([](const shape& s) { return s.min_lens(); }))
+                MIGRAPHX_THROW(prefix() + "Min dynamic dimensions do not match");
        return *this;
    }