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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

/* Copyright © 2023-2024 Apple Inc.                   */
/*                                                    */
/* This file is auto-generated. Do not edit manually. */
/*                                                    */

#include "mlx/c/fft.h"

#include "mlx/c/mlx.h"
#include "mlx/c/private/array.h"
#include "mlx/c/private/closure.h"
#include "mlx/c/private/future.h"
#include "mlx/c/private/io.h"
#include "mlx/c/private/map.h"
#include "mlx/c/private/stream.h"
#include "mlx/c/private/string.h"
#include "mlx/c/private/utils.h"

extern "C" mlx_array mlx_fft_fft(mlx_array a, int n, int axis, mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::fft(a->ctx, n, axis, s->ctx));
}
extern "C" mlx_array mlx_fft_fft2(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::fft2(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}
extern "C" mlx_array mlx_fft_fftn(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::fftn(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}
extern "C" mlx_array mlx_fft_ifft(mlx_array a, int n, int axis, mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::ifft(a->ctx, n, axis, s->ctx));
}
extern "C" mlx_array mlx_fft_ifft2(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::ifft2(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}
extern "C" mlx_array mlx_fft_ifftn(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::ifftn(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}
extern "C" mlx_array mlx_fft_irfft(mlx_array a, int n, int axis, mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::irfft(a->ctx, n, axis, s->ctx));
}
extern "C" mlx_array mlx_fft_irfft2(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::irfft2(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}
extern "C" mlx_array mlx_fft_irfftn(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::irfftn(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}
extern "C" mlx_array mlx_fft_rfft(mlx_array a, int n, int axis, mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::rfft(a->ctx, n, axis, s->ctx));
}
extern "C" mlx_array mlx_fft_rfft2(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::rfft2(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}
extern "C" mlx_array mlx_fft_rfftn(
    mlx_array a,
    const int* n,
    size_t num_n,
    const int* axes,
    size_t num_axes,
    mlx_stream s) {
  RETURN_MLX_C_ARRAY(mlx::core::fft::rfftn(
      a->ctx,
      MLX_CPP_INTVEC(n, num_n),
      MLX_CPP_INTVEC(axes, num_axes),
      s->ctx));
}