/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include <executorch/kernels/portable/cpu/pattern/pattern.h>
#include <executorch/runtime/kernel/kernel_includes.h>
#include <cmath>

namespace torch {
namespace executor {
namespace native {

Tensor& isinf_out(KernelRuntimeContext& ctx, const Tensor& in, Tensor& out) {
  // Lambda is syntactic sugar needed to workaround compilation on some older
  // non-compatible distros where isnan is returning int rather than bool
  return internal::unary_ufunc_realhb_to_bool(
      [](double x) -> bool { return std::isinf(x); }, ctx, in, out);
}

} // namespace native
} // namespace executor
} // namespace torch