mlx-sys 0.1.0-alpha

Rust bindings for mlx
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

#include "mlx-cxx/dtype.hpp"

namespace mlx_cxx {
    mlx::core::Dtype dtype_new(DtypeVal val, uint8_t size) {
        return mlx::core::Dtype(val, size);
    }

    mlx::core::Dtype dtype_bool_() {
        return mlx::core::bool_;
    }

    mlx::core::Dtype dtype_uint8() {
        return mlx::core::uint8;
    }

    mlx::core::Dtype dtype_uint16() {
        return mlx::core::uint16;
    }

    mlx::core::Dtype dtype_uint32() {
        return mlx::core::uint32;
    }

    mlx::core::Dtype dtype_uint64() {
        return mlx::core::uint64;
    }

    mlx::core::Dtype dtype_int8() {
        return mlx::core::int8;
    }

    mlx::core::Dtype dtype_int16() {
        return mlx::core::int16;
    }

    mlx::core::Dtype dtype_int32() {
        return mlx::core::int32;
    }

    mlx::core::Dtype dtype_int64() {
        return mlx::core::int64;
    }

    mlx::core::Dtype dtype_float16() {
        return mlx::core::float16;
    }

    mlx::core::Dtype dtype_float32() {
        return mlx::core::float32;
    }

    mlx::core::Dtype dtype_bfloat16() {
        return mlx::core::bfloat16;
    }

    mlx::core::Dtype dtype_complex64() {
        return mlx::core::complex64;
    }

    std::unique_ptr<std::string> dtype_to_array_protocol(const mlx::core::Dtype& dtype) {
        return std::make_unique<std::string>(mlx::core::dtype_to_array_protocol(dtype));
    }
}