static int aten_to_dtype()

in src/cc/actorpool.cc [530:555]

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

  static int aten_to_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_INT64;
      case at::kInt:
        return NPY_INT32;
      case at::kShort:
        return NPY_INT16;
      case at::kChar:
        return NPY_INT8;
      case at::kByte:
        return NPY_UINT8;
      case at::kBool:
        return NPY_BOOL;
      default: {
        std::string what = "Got unsupported ScalarType ";
        throw py::value_error(what + at::toString(scalar_type));
      }
    }
  }