mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 01:41:17 +08:00
15 lines
305 B
C++
15 lines
305 B
C++
// 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);
|
|
}
|