Method init_

mlx/data/Array.cpp:69–111

}

void Array::init_(
    ArrayType type,
    const std::vector<int64_t>& shape,
    std::shared_ptr<void> data) {
  shape_ = shape;
  type_ = type;
  switch (type_) {
    case UInt8:
      itemsize_ = sizeof(unsigned char);
      break;
    case Int32:
      itemsize_ = sizeof(int32_t);
      break;
    case Int64:
      itemsize_ = sizeof(int64_t);
      break;
    case Float:
      itemsize_ = sizeof(float);
      break;
    case Double:
      itemsize_ = sizeof(double);
      break;
    case Int8:
      itemsize_ = sizeof(char);
      break;
    case Any:
      itemsize_ = 0;
      if (size() != 0) {
        throw std::runtime_error(
            "Array: cannot create a tensor of undetermined type");
      }
      break;
    default:
      throw std::runtime_error("Array: internal error: unknown type");
  }
  if (data) {
    data_ = data;
  } else {
    if (size() > 0) {
      data_ = std::shared_ptr<void>(std::malloc(size() * itemsize_), std::free);
    }
  }
}

Array::Array(const std::string& str) {
  init_(ArrayType::Int8, {static_cast<int64_t>(str.size())});