Skip to content

Commit

Permalink
chore: remove dbg calls
Browse files Browse the repository at this point in the history
  • Loading branch information
polvalente committed Jul 4, 2024
1 parent 9e7e6de commit ca977a7
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions exla/lib/exla/defn.ex
Original file line number Diff line number Diff line change
Expand Up @@ -647,12 +647,12 @@ defmodule EXLA.Defn do
%{client: %EXLA.Client{platform: :host}, builder: %Function{}} = state,
cache
) do
dbg({type_kind})
# We match only on platform: :host for MLIR, as we want to support
# eigh-on-cpu as a custom call only in this case
{tensor, cache} = recur_operator(tensor, state, cache) |> unwrap_single_tensor!()

# convert to float and ensure that we're either using f32 or f64
# convert to float and ensure that we're either using f32 or f64, because Eigen
# only supports f32 and f64 easily.
out_type = Nx.Type.merge(Nx.Type.to_floating(eigenvecs_expr.type), {:f, 32})

tensor =
Expand All @@ -669,8 +669,6 @@ defmodule EXLA.Defn do
expr_to_typespec(%{eigenvals_expr | type: out_type})
)

dbg(eigenvecs)

{[to_type(eigenvecs, eigenvecs_expr.type), to_type(eigenvals, eigenvals_expr.type)], cache}
end

Expand Down

0 comments on commit ca977a7

Please sign in to comment.