2024-04-09 23:50:36 +08:00
|
|
|
# Copyright © 2023-2024 Apple Inc.
|
2023-12-01 03:12:53 +08:00
|
|
|
|
2023-11-30 02:30:41 +08:00
|
|
|
from setuptools import setup
|
|
|
|
|
|
2023-12-09 03:31:47 +08:00
|
|
|
from mlx import extension
|
|
|
|
|
|
2023-11-30 02:30:41 +08:00
|
|
|
if __name__ == "__main__":
|
|
|
|
|
setup(
|
|
|
|
|
name="mlx_sample_extensions",
|
|
|
|
|
version="0.0.0",
|
|
|
|
|
description="Sample C++ and Metal extensions for MLX primitives.",
|
2024-04-09 23:50:36 +08:00
|
|
|
ext_modules=[extension.CMakeExtension("mlx_sample_extensions._ext")],
|
2023-11-30 02:30:41 +08:00
|
|
|
cmdclass={"build_ext": extension.CMakeBuild},
|
|
|
|
|
packages=["mlx_sample_extensions"],
|
|
|
|
|
package_data={"mlx_sample_extensions": ["*.so", "*.dylib", "*.metallib"]},
|
|
|
|
|
zip_safe=False,
|
2023-12-07 23:29:17 +08:00
|
|
|
python_requires=">=3.8",
|
2023-11-30 02:30:41 +08:00
|
|
|
)
|
