[mlir] Add assertion to prevent sending MLIR dynamic shapes (#2299)

a3cf9951 · Krzysztof Drewniak · GitHub · 94fcfc02 · a3cf9951
Unverified Commit a3cf9951 authored Oct 06, 2023 by Krzysztof Drewniak Committed by GitHub Oct 06, 2023
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

src/targets/gpu/mlir.cpp src/targets/gpu/mlir.cpp +4 -1

No files found.
--- a/src/targets/gpu/mlir.cpp
+++ b/src/targets/gpu/mlir.cpp
@@ -320,7 +320,10 @@ struct mlir_program
    MlirType make_tensor(const shape& s) const
    {
-        assert(s.standard());
+        if(not s.standard())
+            MIGRAPHX_THROW("MLIR expects all tensors to be in standard shape");
+        if(s.dynamic())
+            MIGRAPHX_THROW("MLIR does not support dynamic shapes");
        std::vector<int64_t> lens(s.lens().begin(), s.lens().end());
        return mlirRankedTensorTypeGet(
            lens.size(), lens.data(), make_type(s.type()), mlirAttributeGetNull());