c10/core/SymFloat.cpp

#include <c10/core/SymFloat.h>
#include <c10/core/SymFloatNodeImpl.h>
#include <array>

namespace c10 {

SymFloatNode SymFloat::toSymFloatNodeImpl() const {
  TORCH_CHECK(is_symbolic());
  return SymFloatNode::reclaim_copy(toSymFloatNodeImplUnowned());
}

c10::SymFloat SymFloat::toSymFloat(SymFloatNode sin_sp) {
  return c10::SymFloat(std::move(sin_sp));
}

std::ostream& operator<<(std::ostream& os, SymFloat s) {
  if (s.is_symbolic()) {
    os << s.toSymFloatNodeImpl()->str();
  } else {
    os << s.as_float_unchecked();
  }
  return os;
}

} // namespace c10