mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 17:31:16 +08:00
17 lines
580 B
Python
17 lines
580 B
Python
from mlx import extension
|
|
from setuptools import setup
|
|
|
|
if __name__ == "__main__":
|
|
setup(
|
|
name="mlx_sample_extensions",
|
|
version="0.0.0",
|
|
description="Sample C++ and Metal extensions for MLX primitives.",
|
|
ext_modules=[extension.CMakeExtension("mlx_sample_extensions")],
|
|
cmdclass={"build_ext": extension.CMakeBuild},
|
|
packages=["mlx_sample_extensions"],
|
|
package_dir={"": "."},
|
|
package_data={"mlx_sample_extensions": ["*.so", "*.dylib", "*.metallib"]},
|
|
zip_safe=False,
|
|
python_requires=">=3.7",
|
|
)
|