mlx-sys 0.0.8

Low-level interface and binding generation for the mlx library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

/* Copyright © 2023-2024 Apple Inc. */

#ifndef MLX_ARRAY_PRIVATE_H
#define MLX_ARRAY_PRIVATE_H

#include "mlx/c/array.h"
#include "mlx/c/private/object.h"
#include "mlx/mlx.h"

struct mlx_array_ : mlx_object_ {
  mlx_array_(mlx::core::array ctx) : mlx_object_(), ctx(ctx){};
  virtual mlx_string_* tostring() override;
  mlx::core::array ctx;
};

struct mlx_vector_array_ : mlx_object_ {
  mlx_vector_array_() : mlx_object_(){};
  mlx_vector_array_(const std::vector<mlx::core::array>& ctx)
      : mlx_object_(), ctx(ctx){};
  mlx_vector_array_(std::vector<mlx::core::array>&& ctx)
      : mlx_object_(), ctx(std::move(ctx)){};
  mlx_vector_array_(const std::pair<mlx::core::array, mlx::core::array>& ctx)
      : mlx_object_(), ctx({ctx.first, ctx.second}){};
  mlx_vector_array_(
      const std::tuple<mlx::core::array, mlx::core::array, mlx::core::array>&
          ctx)
      : mlx_object_(),
        ctx({std::get<0>(ctx), std::get<1>(ctx), std::get<2>(ctx)}){};
  virtual mlx_string_* tostring() override;
  std::vector<mlx::core::array> ctx;
};

struct mlx_vector_vector_array_ : mlx_object_ {
  mlx_vector_vector_array_() : mlx_object_(){};
  mlx_vector_vector_array_(
      const std::vector<std::vector<mlx::core::array>>& ctx)
      : mlx_object_(), ctx(ctx){};
  mlx_vector_vector_array_(std::vector<std::vector<mlx::core::array>>&& ctx)
      : mlx_object_(), ctx(std::move(ctx)){};
  mlx_vector_vector_array_(const std::pair<
                           std::vector<mlx::core::array>,
                           std::vector<mlx::core::array>>& ctx)
      : mlx_object_(), ctx({ctx.first, ctx.second}){};
  virtual mlx_string_* tostring() override;
  std::vector<std::vector<mlx::core::array>> ctx;
};

#endif