Refactor module debugging

Author: jonatanklosko

exla/lib/exla/mlir/module.ex

@@ -98,7 +98,8 @@ defmodule EXLA.MLIR.Module do
         do: -1,
         else: Keyword.get(options, :device_id, client.default_device_id)
 
-    # module.ref |> EXLA.NIF.mlir_module_to_string() |> elem(1) |> IO.puts()
+    # Uncomment to debug the module MLIR source
+    # module |> as_string() |> IO.puts()
 
     ref =
       EXLA.NIF.mlir_compile(
@@ -126,7 +127,7 @@ defmodule EXLA.MLIR.Module do
   Returns a human-readable representation of the module using MLIR
   syntax.
   """
-  def to_string(module = %__MODULE__{}) do
+  def as_string(module = %__MODULE__{}) do
     EXLA.NIF.mlir_module_to_string(module.ref) |> unwrap!()
   end