mlx-sys 0.0.10-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

// Copyright © 2023 Apple Inc.

#include <pybind11/pybind11.h>

#include "mlx/backend/metal/metal.h"

namespace py = pybind11;

using namespace mlx::core;

void init_metal(py::module_& m) {
  py::module_ metal = m.def_submodule("metal", "mlx.metal");
  metal.def("is_available", &metal::is_available);
  metal.def(
      "cache_enabled",
      &metal::cache_enabled,
      "check if metal buffer cache is enabled, default is true");
  metal.def(
      "set_cache_enabled",
      &metal::set_cache_enabled,
      "enable or disable metal buffer cache");
}