mirtal-sys 0.1.0

Private native bridge for mirtal
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 "mirtal-sys/src/lib.rs.h"

#include <optional>
#include <utility>

namespace mirtal {
mx::array rope_native(
    const mx::array& input,
    std::int32_t dimensions,
    bool traditional,
    std::optional<float> base,
    float scale,
    std::int32_t offset,
    std::optional<mx::array> frequencies,
    const mx::Stream& stream) {
  return mx::fast::rope(
      input,
      dimensions,
      traditional,
      base,
      scale,
      offset,
      frequencies,
      stream);
}

std::shared_ptr<Array> rope(
    const Array& input,
    const NativeRopeOptions& options,
    const Stream& stream) {
  auto optional_base = options.has_base
      ? std::make_optional(options.base)
      : std::optional<float>{};
  return std::make_shared<Array>(rope_native(
      input.value,
      options.dimensions,
      options.traditional,
      optional_base,
      options.scale,
      options.offset,
      std::nullopt,
      stream.value));
}

std::shared_ptr<Array> rope_with_frequencies(
    const Array& input,
    std::int32_t dimensions,
    bool traditional,
    const Array& frequencies,
    std::int32_t offset,
    const Stream& stream) {
  return std::make_shared<Array>(rope_native(
      input.value,
      dimensions,
      traditional,
      std::nullopt,
      1.0f,
      offset,
      frequencies.value,
      stream.value));
}

}  // namespace mirtal