Do not require scalar for memref set value (#101)

Author: hunhoffe

mlir/extras/dialects/ext/memref.py

@@ -172,6 +172,9 @@ def __setitem__(self, idx, val):
                 idx[i] = constant(d, index=True, loc=loc)
 
         if all(isinstance(d, Scalar) for d in idx) and len(idx) == len(self.shape):
+            if isinstance(val, (int, float)):
+                # TODO: this is an unchecked conversion
+                val = Scalar(val, dtype=self.dtype)
             assert isinstance(val, Scalar), "coordinate insert requires scalar element"
             store(val, self, idx, loc=loc)
         else: