Scalar.cpp

#include "ATen/Scalar.h"

#include <TH/TH.h>

#include "ATen/Tensor.h"
#include "ATen/Context.h"

namespace at {

Tensor Scalar::toTensor() const {
  if (Tag::HAS_t == tag) {
    return Tensor(t);
  } else if (Tag::HAS_d == tag) {
    return CPU(kDouble).scalarTensor(*this);
  } else {
    assert(Tag::HAS_i == tag);
    return CPU(kLong).scalarTensor(*this);
  }
}

template<> Half convert(double f) {
  float t = static_cast<float>(f);
  Half h;
  TH_float2halfbits(&t,&h.x);
  return h;
}
template<> double convert(Half f) {
  float t;
  TH_halfbits2float(&f.x,&t);
  return t;
}
template<> Half convert(int64_t f) {
  return convert<Half,double>(static_cast<double>(f));
}
template<> int64_t convert(Half f) {
  return static_cast<int64_t>(convert<double,Half>(f));
}


#ifdef AT_CUDA_ENABLED
template<> half convert(double d) {

#if CUDA_VERSION < 9000
  return half {convert<Half,double>(d).x};
#else
  __half_raw raw;
  raw.x = convert<Half,double>(d).x;
  return half {raw};
#endif
}
#endif

}