Sha256: 7596a7ab6fc64371ec093bd80d3b02a734bc55e9864806b7a5fc1399719a8218

Contents?: true

Size: 1.18 KB

Versions: 6

Compression:

Stored size: 1.18 KB

Contents

#pragma once

#include <rice/rice.hpp>
#include <rice/stl.hpp>

// TODO find better place
inline void handle_error(torch::Error const & ex) {
  throw Rice::Exception(rb_eRuntimeError, ex.what_without_backtrace());
}

// keep THP prefix for now to make it easier to compare code

extern VALUE THPVariableClass;

inline VALUE THPUtils_internSymbol(const std::string& str) {
  return Rice::Symbol(str);
}

inline std::string THPUtils_unpackSymbol(VALUE obj) {
  Check_Type(obj, T_SYMBOL);
  obj = rb_funcall(obj, rb_intern("to_s"), 0);
  return std::string(RSTRING_PTR(obj), RSTRING_LEN(obj));
}

inline std::string THPUtils_unpackString(VALUE obj) {
  Check_Type(obj, T_STRING);
  return std::string(RSTRING_PTR(obj), RSTRING_LEN(obj));
}

inline bool THPUtils_checkSymbol(VALUE obj) {
  return SYMBOL_P(obj);
}

inline bool THPUtils_checkIndex(VALUE obj) {
  return FIXNUM_P(obj);
}

inline bool THPUtils_checkScalar(VALUE obj) {
  return FIXNUM_P(obj) || RB_FLOAT_TYPE_P(obj) || RB_TYPE_P(obj, T_COMPLEX);
}

inline bool THPVariable_Check(VALUE obj) {
  return rb_obj_is_kind_of(obj, THPVariableClass);
}

inline bool THPVariable_CheckExact(VALUE obj) {
  return rb_obj_is_instance_of(obj, THPVariableClass);
}

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
torch-rb-0.9.0 ext/torch/utils.h
torch-rb-0.8.3 ext/torch/utils.h
torch-rb-0.8.2 ext/torch/utils.h
torch-rb-0.8.1 ext/torch/utils.h
torch-rb-0.8.0 ext/torch/utils.h
torch-rb-0.7.0 ext/torch/utils.h