2023-12-01 03:12:53 +08:00
|
|
|
# Copyright © 2023 Apple Inc.
|
|
|
|
|
|
2023-11-30 02:30:41 +08:00
|
|
|
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",
|
|
|
|
|
)
|
