feat: add msg/reply for sending an execution to a liveview

Author: polvalente

liveview_native/live_nx_iree/lib/live_nx_iree_web/live/home_live/home_live.ex

@@ -15,7 +15,7 @@ defmodule LiveNxIREEWeb.HomeLive do
 
     dbg(self())
 
-    socket = assign(socket, bytecode: nil, function_signature: nil)
+    socket = assign(socket, bytecode: nil, function_signature: nil, device: nil)
 
     {:ok, socket}
   end
@@ -43,7 +43,7 @@ defmodule LiveNxIREEWeb.HomeLive do
   # end
 
   @impl true
-  def handle_info({:nx, :execute, function, input_templates, target_device}, socket) do
+  def handle_info({:nx, :execute, function, input_templates, target_device, reply_to_pid}, socket) do
     fun =
       case function do
         {m, f, a} ->
@@ -55,10 +55,13 @@ defmodule LiveNxIREEWeb.HomeLive do
           """
       end
 
-    backend_flag =
+    {backend_flag, runtime_device} =
       case target_device do
-        :metal -> "--iree-hal-target-backends=metal-spirv"
-        :cpu -> "--iree-hal-target-backends=llvm-cpu"
+        :metal ->
+          {"--iree-hal-target-backends=metal-spirv", "metal://default"}
+
+        :cpu ->
+          {"--iree-hal-target-backends=llvm-cpu", "local-sync://default"}
       end
 
     compiler_flags = [
@@ -75,10 +78,19 @@ defmodule LiveNxIREEWeb.HomeLive do
       |> assign(:bytecode, Base.encode64(bytecode))
       |> assign(:output_container, output_container)
       |> assign(:function_signature, get_signature(function, input_templates, output_container))
+      |> assign(:device, runtime_device)
+      |> assign(:reply_to_pid, reply_to_pid)
 
     {:noreply, socket}
   end
 
+  @impl true
+  def handle_event("nx-executed", params, socket) do
+    send(socket.assigns.reply_to_pid, {:nx, :executed, params})
+
+    {:noreply, assign(socket, :reply_to_pid, nil)}
+  end
+
   defp get_signature({mod, fun, _a}, input_templates, output_container) do
     "#{inspect(mod)}.#{fun}(#{to_flat_type(input_templates)}) -> #{to_flat_type(output_container)}"
   end

liveview_native/live_nx_iree/lib/live_nx_iree_web/live/swiftui/home_live.swiftui.neex

@@ -1 +1 @@
-<NxFunction signature={@function_signature} bytecode={@bytecode}/>
+<NxFunction on-execution="nx-executed" signature={@function_signature} bytecode={@bytecode} device={@device} />

liveview_native/live_nx_iree/mix.exs

@@ -19,7 +19,7 @@ defmodule LiveNxIREE.MixProject do
   def application do
     [
       mod: {LiveNxIREE.Application, []},
-      extra_applications: [:logger, :runtime_tools]
+      extra_applications: [:logger, :runtime_tools, :observer]
     ]
   end
 

liveview_native/live_nx_iree/native/swiftui/LiveNxIREE.xcodeproj/project.xcworkspace/xcuserdata/paulo.valente.xcuserdatad/UserInterfaceState.xcuserstate

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Bucket
+   uuid = "B200013E-0754-4DEE-8498-6AF0A1807EEA"
+   type = "1"
+   version = "2.0">
+</Bucket>

liveview_native/live_nx_iree/native/swiftui/LiveNxIREE.xcodeproj/xcuserdata/paulo.valente.xcuserdatad/xcdebugger/Breakpoints_v2.xcbkptlist

@@ -13,11 +13,32 @@ struct NxFunctionView<Root: RootRegistry>: View {
     @_documentation(visibility: public)
     @LiveAttribute("bytecode") private var bytecode: String? = nil
     @LiveAttribute("signature") private var signature: String? = nil
+    @LiveAttribute("device") private var device: String? = nil
+    @LiveAttribute("trigger") private var trigger: Bool = false
+    @Event("on-execution", type: "change") private var change
 
     var body: some View {
-        if signature != nil {
-            Text(signature!)
-                .padding()
+        VStack {
+            if signature != nil {
+                Text(signature!)
+                    .padding()
+            }
+        }
+        .onAppear() {
+            run()
+        }
+        .onChange(of: bytecode) {
+            run()  // Run the function when bytecode changes
         }
     }
+    
+    private func run() {
+        if bytecode == nil {
+            return
+        }
+       // Custom logic that should run when execute is set to true
+       print("Executing function \(signature ?? "None") on device: \(device ?? "None")")
+        
+       change(value: "Sending something back")
+   }
 }