mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 17:31:16 +08:00
b63ef10a7f
* start to fix extensions * mostly fixed extensions * fix extension build * couple more nits
20 lines
628 B
Python
20 lines
628 B
Python
# Copyright © 2023-2024 Apple Inc.
|
|
|
|
from setuptools import setup
|
|
|
|
from mlx import extension
|
|
|
|
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._ext")],
|
|
cmdclass={"build_ext": extension.CMakeBuild},
|
|
packages=["mlx_sample_extensions"],
|
|
package_data={"mlx_sample_extensions": ["*.so", "*.dylib", "*.metallib"]},
|
|
extras_require={"dev": []},
|
|
zip_safe=False,
|
|
python_requires=">=3.8",
|
|
)
|