minor fix (#62)

Author: makslevental

mlir/extras/dialects/ext/tensor.py

@@ -26,9 +26,7 @@
 S = ShapedType.get_dynamic_size()
 
 
-def empty(
-    *sizes: Sequence[Union[int, Value]], element_type: Type = None, loc=None, ip=None
-):
+def empty(*sizes: Union[int, Value], element_type: Type = None, loc=None, ip=None):
     if loc is None:
         loc = get_user_code_loc()
     if element_type is None:
@@ -608,12 +606,7 @@ def _insert_slice(
 ):
     if loc is None:
         loc = get_user_code_loc()
-
-    if isinstance(source, Scalar):
-        source = expand_dims(source, (0,), loc=loc, ip=ip)
-
     indexer = _indices_to_indexer(idx, dest.shape)
-
     if indexer.is_constant():
         assert (
             indexer.static_sizes() == source.shape

mlir/extras/dialects/ext/transform.py

@@ -8,25 +8,20 @@
 from ...util import get_user_code_loc
 from ....dialects import pdl
 from ....dialects import transform
-from ....dialects._ods_common import (
-    _dispatch_mixed_values,
-)
-from ....dialects._ods_common import get_op_result_or_op_results
+from ....dialects._ods_common import _dispatch_mixed_values, get_op_result_or_op_results
 from ....dialects._structured_transform_ops_gen import (
     TileUsingForallOp,
     MatchOp,
 )
 from ....dialects.transform import *
+from ....dialects.transform import AnyOpType, AnyValueType, OperationType
 from ....dialects.transform.structured import TileUsingForOp
+from ....dialects.transform.loop import LoopUnrollOp
 from ....ir import Type, Operation, StringAttr, Attribute, Value
 
 transform_fully_qualified_name = transform.__spec__.name
 
 
-def create_simple_namespace(name):
-    return SimpleNamespace(__name__=name)
-
-
 # transform.apply_patterns is both a namespace and an op...
 delattr(transform, "apply_patterns")
 
@@ -57,7 +52,8 @@ def create_simple_namespace(name):
 
             for i, n in enumerate(namespaces[1:-1]):
                 if not hasattr(simple_namespace, n):
-                    # dumb: without the prefix, this somehow always names the modules "mlir.dialect.module.transform.<n>" instead of suffixing
+                    # dumb: without the prefix, this somehow always names the modules
+                    # "mlir.dialect.module.transform.<n>" instead of suffixing
                     setattr(
                         simple_namespace,
                         n,