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

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

#include "mlx/c/ioutils.h"
#include "mlx/c/private/io.h"
#include "mlx/c/private/map.h"
#include "mlx/c/private/string.h"
#include "mlx/c/private/utils.h"

mlx_string_* mlx_safetensors_::tostring() {
  MLX_TRY_CATCH(return new mlx_string_("mlx_safetensors"), return nullptr);
}

extern "C" mlx_safetensors mlx_safetensors_new() {
  MLX_TRY_CATCH(return new mlx_safetensors_(), return nullptr);
}

/**
 * Returns the data map.
 */
extern "C" mlx_map_string_to_array mlx_safetensors_data(mlx_safetensors st) {
  MLX_TRY_CATCH(mlx_retain(st->data); return st->data, return nullptr);
}
/**
 * Returns the metadata map.
 */
extern "C" mlx_map_string_to_string mlx_safetensors_metadata(
    mlx_safetensors st) {
  MLX_TRY_CATCH(mlx_retain(st->metadata); return st->metadata, return nullptr);
}