mlx/python/src/metal.cpp

// 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");
}