c10/core/SymNodeImpl.h

#pragma once

#include <c10/macros/Macros.h>
#include <c10/util/Exception.h>
#include <c10/util/intrusive_ptr.h>
#include <memory>
#include <mutex>
#include <vector>

namespace c10 {

class SymNodeImpl;
using SymNode = c10::intrusive_ptr<SymNodeImpl>;

class C10_API SymNodeImpl : public c10::intrusive_ptr_target {
 public:
  virtual ~SymNodeImpl(){};

  template <typename T>
  c10::intrusive_ptr<T> dyn_cast() const {
    return c10::intrusive_ptr<T>::reclaim_copy(dynamic_cast<T*>(this));
  }

  // these could be pure virtual when we implement LTC versions
  virtual bool is_int() {
    TORCH_CHECK(false, "NYI");
  };
  virtual bool is_float() {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode add(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode sub(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode mul(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode truediv(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode pow(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode floordiv(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode mod(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode eq(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode ne(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode gt(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode lt(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode le(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode ge(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode ceil() {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode floor() {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode neg() {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode min(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode max(const SymNode& other) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode clone() {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode sym_float() {
    TORCH_CHECK(false, "NYI");
  }
  virtual SymNode wrap_int(int64_t num) {
    TORCH_CHECK(false, "NYI");
  };
  virtual SymNode wrap_float(double num) {
    TORCH_CHECK(false, "NYI");
  };
  virtual int64_t guard_int(const char* file, int64_t line) {
    TORCH_CHECK(false, "NYI");
  };
  virtual double guard_float(const char* file, int64_t line) {
    TORCH_CHECK(false, "NYI");
  };
  virtual int64_t int_() {
    TORCH_CHECK(false, "NYI");
  };
  virtual bool bool_() {
    TORCH_CHECK(false, "NYI");
  };
  virtual std::string str() {
    TORCH_CHECK(false, "NYI");
  };
  std::ostream& operator<<(std::ostream& os) {
    os << str();
    return os;
  };
};

} // namespace c10