update callbacks

mimiquate · Jan 16, 2025 · 6b9c873 · 6b9c873
1 parent a9118b5
commit 6b9c873
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lib/candlex/backend.ex b/lib/candlex/backend.ex
@@ -1041,7 +1041,8 @@ defmodule Candlex.Backend do
         :count_leading_zeros,
         :imag,
         :population_count,
-        :real
+        :real,
+        :to_pointer
       ] do
     @impl true
     def unquote(op)(_out, _tensor) do
@@ -1056,15 +1057,16 @@ defmodule Candlex.Backend do
         :lu,
         :product,
         :qr,
-        :sort
+        :sort,
+        :stack
       ] do
     @impl true
     def unquote(op)(_out, _tensor, _) do
       raise "unsupported Candlex.Backend.#{unquote(op)} function"
     end
   end
 
-  for op <- [:map, :triangular_solve, :window_min, :window_product] do
+  for op <- [:triangular_solve, :window_min, :window_product] do
     @impl true
     def unquote(op)(_out, _tensor, _, _) do
       raise "unsupported Candlex.Backend.#{unquote(op)} function"
@@ -1081,6 +1083,7 @@ defmodule Candlex.Backend do
   end
 
   for op <- [
+        :from_pointer,
         :indexed_put,
         :reduce
       ] do