[mlir][python] Actually set UseLocalScope printing flag

[mabrowning]

Summary

The useLocalScope printing flag has been passed around between pybind methods, but doesn't actually enable the corresponding printing flag.

[stellaraccident]

Thanks - good catch. Do you have commit access to land?

[mabrowning]

In D127907#3587153, @stellaraccident wrote:

Thanks - good catch. Do you have commit access to land?

I don't think so, this is my first contribution.

[stellaraccident]

Thanks for the contribution. I'll land it at my next opportunity, then.

[mabrowning]

In D127907#3587189, @stellaraccident wrote:

Thanks for the contribution. I'll land it at my next opportunity, then.

Sorry, looks like I need to rebase my patch; I had taken it from a slightly version of our internal copy of the repo. The operation.py test slightly changes.

[mabrowning]

"rebased" onto latest main, updated test due to conflict.

mlir/lib/Bindings/Python/IRCore.cpp

     mlirOpPrintingFlagsEnableDebugInfo(flags, /*prettyForm=*/prettyDebugInfo);
   if (printGenericOpForm)
     mlirOpPrintingFlagsPrintGenericOpForm(flags);
+  if (useLocalScope)
+    mlirOpPrintingFlagsUseLocalScope(flags);
 
   PyFileAccumulator accum(fileObject, binary);
   py::gil_scoped_release();

mlir/test/python/ir/operation.py

   ctx = Context()
   module = Module.parse(r"""
     func @f1(%arg0: i32) -> i32 {
-      %0 = constant dense<[1, 2, 3, 4]> : tensor<4xi32>
+      %0 = constant dense<[1, 2, 3, 4]> : tensor<4xi32> loc("nom")
       return %arg0 : i32
     }
   """, ctx)

   print(bytes_value.__class__)
   print(bytes_value)
 
+  # Test get_asm local_scope.
+  # CHECK: constant opaque<"_", "0xDEADBEEF"> : tensor<4xi32> loc("nom")
+  module.operation.print(large_elements_limit=2, enable_debug_info=True,
+      use_local_scope=True)
+
   # Test get_asm with options.
   # CHECK: value = opaque<"_", "0xDEADBEEF"> : tensor<4xi32>
   # CHECK: "std.return"(%arg0) : (i32) -> () -:4:7