int aten_to_numpy_dtype()

in congestion_control/Utils.cpp [17:41]

• 25 lines of code
• 10 McCabe index (conditional complexity)

int aten_to_numpy_dtype(const at::ScalarType scalar_type) {
  switch (scalar_type) {
  case at::kDouble:
    return NPY_DOUBLE;
  case at::kFloat:
    return NPY_FLOAT;
  case at::kHalf:
    return NPY_HALF;
  case at::kLong:
    return NPY_LONG;
  case at::kInt:
    return NPY_INT;
  case at::kShort:
    return NPY_SHORT;
  case at::kChar:
    return NPY_BYTE;
  case at::kByte:
    return NPY_UBYTE;
  case at::kBool:
    return NPY_BOOL;
  default:
    throw std::runtime_error(
        folly::sformat("Unsupported ScalarType: {}", toString(scalar_type)));
  }
}