Validate memref element type in mapArrayType (#156)

Hardcode84 · web-flow · commit 4ada91223c6e · 2022-02-21T13:25:42.000+03:00
diff --git a/numba_dpcomp/numba_dpcomp/mlir_compiler/lib/pipelines/plier_to_linalg.cpp b/numba_dpcomp/numba_dpcomp/mlir_compiler/lib/pipelines/plier_to_linalg.cpp
@@ -179,9 +179,11 @@ static mlir::Type mapArrayType(mlir::MLIRContext &ctx,
         desc->layout == ArrayLayout::A) {
       if (auto type =
               conveter.convertType(plier::PyType::get(&ctx, desc->name))) {
-        llvm::SmallVector<int64_t> shape(desc->dims,
-                                         mlir::ShapedType::kDynamicSize);
-        return mlir::MemRefType::get(shape, type);
+        if (mlir::BaseMemRefType::isValidElementType(type)) {
+          llvm::SmallVector<int64_t> shape(desc->dims,
+                                           mlir::ShapedType::kDynamicSize);
+          return mlir::MemRefType::get(shape, type);
+        }
       }
     }
   }

Original file line number	Diff line number	Diff line change
`@@ -179,9 +179,11 @@ static mlir::Type mapArrayType(mlir::MLIRContext &ctx,`
`179`	`179`	`desc->layout == ArrayLayout::A) {`
`180`	`180`	`if (auto type =`
`181`	`181`	`conveter.convertType(plier::PyType::get(&ctx, desc->name))) {`
`182`		`- llvm::SmallVector<int64_t> shape(desc->dims,`
`183`		`- mlir::ShapedType::kDynamicSize);`
`184`		`- return mlir::MemRefType::get(shape, type);`
	`182`	`+ if (mlir::BaseMemRefType::isValidElementType(type)) {`
	`183`	`+ llvm::SmallVector<int64_t> shape(desc->dims,`
	`184`	`+ mlir::ShapedType::kDynamicSize);`
	`185`	`+ return mlir::MemRefType::get(shape, type);`
	`186`	`+ }`
`185`	`187`	`}`
`186`	`188`	`}`
`187`	`189`	`}`