fix

elixir-nx · Aug 31, 2023 · b34738c · b34738c
1 parent e9c4f39
commit b34738c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/exla/lib/exla/computation.ex b/exla/lib/exla/computation.ex
@@ -83,7 +83,7 @@ defmodule EXLA.Computation do
       client,
       arg_shapes,
       # TO-DO (mlir): do not hardcode this single-item tuple output type
-      EXLA.Shape.make_tuple_shape([ret_shape])
+      EXLA.Shape.make_tuple_shape([return_shape])
     )
   end