From 46a39e5b1ff92d93833d9710ef64db207ce274ea Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Thu, 30 Nov 2023 11:12:53 -0800 Subject: [PATCH] copyright + ack --- ACKNOWLEDGMENTS.md | 30 +++++++++++++++++++ LICENSE | 2 +- benchmarks/cpp/autograd.cpp | 2 ++ benchmarks/cpp/compare_devices.cpp | 2 ++ benchmarks/cpp/irregular_strides.cpp | 2 ++ benchmarks/cpp/single_ops.cpp | 2 ++ benchmarks/cpp/time_utils.h | 2 ++ benchmarks/numpy/single_ops.py | 2 ++ benchmarks/numpy/time_utils.py | 2 ++ benchmarks/python/batch_matmul_bench.py | 2 ++ benchmarks/python/blas/bench_gemm.py | 2 ++ benchmarks/python/blas/bench_gemv.py | 2 ++ benchmarks/python/comparative/bench_mlx.py | 2 ++ benchmarks/python/comparative/bench_torch.py | 2 ++ benchmarks/python/comparative/compare.py | 2 ++ benchmarks/python/llama_jax_bench.py | 2 ++ benchmarks/python/llama_mlx_bench.py | 2 ++ benchmarks/python/llama_torch_bench.py | 2 ++ benchmarks/python/single_ops.py | 2 ++ benchmarks/python/time_utils.py | 2 ++ docs/src/conf.py | 2 ++ examples/cpp/linear_regression.cpp | 2 ++ examples/cpp/logistic_regression.cpp | 2 ++ examples/cpp/timer.h | 2 ++ examples/cpp/tutorial.cpp | 2 ++ examples/extensions/axpby/axpby.cpp | 2 ++ examples/extensions/axpby/axpby.h | 2 ++ examples/extensions/axpby/axpby.metal | 2 ++ examples/extensions/bindings.cpp | 2 ++ .../mlx_sample_extensions/__init__.py | 2 ++ examples/extensions/setup.py | 2 ++ examples/python/linear_regression.py | 2 ++ examples/python/logistic_regression.py | 2 ++ mlx/allocator.cpp | 2 ++ mlx/allocator.h | 2 ++ mlx/array.cpp | 2 ++ mlx/array.h | 2 ++ mlx/backend/accelerate/conv.cpp | 2 ++ mlx/backend/accelerate/matmul.cpp | 2 ++ mlx/backend/accelerate/primitives.cpp | 2 ++ mlx/backend/accelerate/reduce.cpp | 2 ++ mlx/backend/accelerate/softmax.cpp | 2 ++ mlx/backend/accelerate/utils.h | 2 ++ mlx/backend/common/arange.h | 2 ++ mlx/backend/common/arg_reduce.cpp | 2 ++ mlx/backend/common/binary.cpp | 2 ++ mlx/backend/common/binary.h | 2 ++ mlx/backend/common/conv.cpp | 2 ++ mlx/backend/common/copy.cpp | 2 ++ mlx/backend/common/copy.h | 2 ++ mlx/backend/common/default_primitives.cpp | 2 ++ mlx/backend/common/erf.cpp | 2 ++ mlx/backend/common/erf.h | 2 ++ mlx/backend/common/fft.cpp | 2 ++ mlx/backend/common/indexing.cpp | 2 ++ mlx/backend/common/load.cpp | 2 ++ mlx/backend/common/primitives.cpp | 2 ++ mlx/backend/common/reduce.cpp | 2 ++ mlx/backend/common/reduce.h | 2 ++ mlx/backend/common/scan.cpp | 2 ++ mlx/backend/common/softmax.cpp | 2 ++ mlx/backend/common/sort.cpp | 2 ++ mlx/backend/common/threefry.cpp | 2 ++ mlx/backend/common/threefry.h | 2 ++ mlx/backend/common/unary.h | 2 ++ mlx/backend/common/utils.h | 2 ++ mlx/backend/metal/allocator.cpp | 2 ++ mlx/backend/metal/allocator.h | 2 ++ mlx/backend/metal/conv.cpp | 2 ++ mlx/backend/metal/copy.cpp | 2 ++ mlx/backend/metal/copy.h | 2 ++ mlx/backend/metal/device.cpp | 2 ++ mlx/backend/metal/device.h | 2 ++ mlx/backend/metal/fft.cpp | 2 ++ mlx/backend/metal/indexing.cpp | 2 ++ mlx/backend/metal/kernels/arange.metal | 2 ++ mlx/backend/metal/kernels/arg_reduce.metal | 2 ++ mlx/backend/metal/kernels/atomic.h | 2 ++ mlx/backend/metal/kernels/bf16.h | 2 ++ mlx/backend/metal/kernels/bf16_math.h | 2 ++ mlx/backend/metal/kernels/binary.metal | 2 ++ mlx/backend/metal/kernels/complex.h | 2 ++ mlx/backend/metal/kernels/conv.metal | 2 ++ mlx/backend/metal/kernels/conv_params.h | 2 ++ mlx/backend/metal/kernels/copy.metal | 2 ++ mlx/backend/metal/kernels/defines.h | 2 ++ mlx/backend/metal/kernels/erf.h | 2 ++ mlx/backend/metal/kernels/gemm.metal | 2 ++ mlx/backend/metal/kernels/gemm/conv.h | 2 ++ mlx/backend/metal/kernels/gemm/gemm.h | 2 ++ mlx/backend/metal/kernels/gemv.metal | 2 ++ mlx/backend/metal/kernels/indexing.metal | 2 ++ mlx/backend/metal/kernels/random.metal | 2 ++ mlx/backend/metal/kernels/reduce.h | 2 ++ mlx/backend/metal/kernels/reduce.metal | 2 ++ mlx/backend/metal/kernels/scan.metal | 2 ++ mlx/backend/metal/kernels/softmax.metal | 2 ++ mlx/backend/metal/kernels/sort.metal | 2 ++ mlx/backend/metal/kernels/unary.metal | 2 ++ mlx/backend/metal/kernels/utils.h | 2 ++ mlx/backend/metal/matmul.cpp | 2 ++ mlx/backend/metal/matmul.h | 2 ++ mlx/backend/metal/metal.cpp | 2 ++ mlx/backend/metal/metal.h | 2 ++ mlx/backend/metal/mps/gemm.h | 2 ++ mlx/backend/metal/primitives.cpp | 2 ++ mlx/backend/metal/reduce.cpp | 2 ++ mlx/backend/metal/scan.cpp | 2 ++ mlx/backend/metal/softmax.cpp | 2 ++ mlx/backend/metal/sort.cpp | 2 ++ mlx/backend/metal/utils.h | 2 ++ mlx/backend/no_metal/allocator.cpp | 2 ++ mlx/backend/no_metal/metal.cpp | 2 ++ mlx/backend/no_metal/primitives.cpp | 2 ++ mlx/device.cpp | 2 ++ mlx/device.h | 2 ++ mlx/dtype.cpp | 2 ++ mlx/dtype.h | 2 ++ mlx/fft.cpp | 2 ++ mlx/fft.h | 2 ++ mlx/graph_utils.cpp | 2 ++ mlx/graph_utils.h | 2 ++ mlx/load.cpp | 2 ++ mlx/load.h | 2 ++ mlx/mlx.h | 2 ++ mlx/ops.cpp | 2 ++ mlx/ops.h | 2 ++ mlx/primitives.cpp | 2 ++ mlx/primitives.h | 2 ++ mlx/random.cpp | 2 ++ mlx/random.h | 2 ++ mlx/scheduler.cpp | 2 ++ mlx/scheduler.h | 2 ++ mlx/stream.h | 2 ++ mlx/transforms.cpp | 2 ++ mlx/transforms.h | 2 ++ mlx/transforms_impl.h | 2 ++ mlx/types/bf16.h | 2 ++ mlx/types/complex.h | 2 ++ mlx/types/fp16.h | 2 ++ mlx/types/half_types.h | 2 ++ mlx/utils.cpp | 2 ++ mlx/utils.h | 2 ++ python/mlx/_reprlib_fix.py | 2 ++ python/mlx/extension.py | 2 ++ python/mlx/nn/__init__.py | 2 ++ python/mlx/nn/layers/__init__.py | 2 ++ python/mlx/nn/layers/activations.py | 2 ++ python/mlx/nn/layers/base.py | 2 ++ python/mlx/nn/layers/containers.py | 2 ++ python/mlx/nn/layers/convolution.py | 2 ++ python/mlx/nn/layers/dropout.py | 2 ++ python/mlx/nn/layers/embedding.py | 2 ++ python/mlx/nn/layers/linear.py | 2 ++ python/mlx/nn/layers/normalization.py | 2 ++ python/mlx/nn/layers/positional_encoding.py | 2 ++ python/mlx/nn/layers/transformer.py | 2 ++ python/mlx/nn/losses.py | 2 ++ python/mlx/nn/utils.py | 2 ++ python/mlx/optimizers.py | 2 ++ python/mlx/utils.py | 2 ++ python/src/array.cpp | 2 ++ python/src/device.cpp | 2 ++ python/src/fft.cpp | 2 ++ python/src/indexing.cpp | 2 ++ python/src/indexing.h | 2 ++ python/src/load.cpp | 2 ++ python/src/load.h | 2 ++ python/src/metal.cpp | 2 ++ python/src/mlx.cpp | 2 ++ python/src/ops.cpp | 2 ++ python/src/random.cpp | 2 ++ python/src/stream.cpp | 2 ++ python/src/transforms.cpp | 2 ++ python/src/utils.h | 2 ++ python/tests/mlx_tests.py | 2 ++ python/tests/test_array.py | 2 ++ python/tests/test_autograd.py | 2 ++ python/tests/test_bf16.py | 2 ++ python/tests/test_blas.py | 2 ++ python/tests/test_conv.py | 2 ++ python/tests/test_device.py | 2 ++ python/tests/test_eval.py | 2 ++ python/tests/test_fft.py | 2 ++ python/tests/test_load.py | 2 ++ python/tests/test_nn.py | 2 ++ python/tests/test_ops.py | 2 ++ python/tests/test_optimizers.py | 2 ++ python/tests/test_random.py | 2 ++ python/tests/test_reduce.py | 2 ++ python/tests/test_tree.py | 2 ++ python/tests/test_vmap.py | 2 ++ setup.py | 2 ++ tests/allocator_tests.cpp | 2 ++ tests/arg_reduce_tests.cpp | 2 ++ tests/array_tests.cpp | 2 ++ tests/autograd_tests.cpp | 2 ++ tests/blas_tests.cpp | 2 ++ tests/creations_tests.cpp | 2 ++ tests/device_tests.cpp | 2 ++ tests/eval_tests.cpp | 2 ++ tests/fft_tests.cpp | 2 ++ tests/graph_optimize_tests.cpp | 2 ++ tests/load_tests.cpp | 2 ++ tests/metal_tests.cpp | 2 ++ tests/ops_tests.cpp | 2 ++ tests/random_tests.cpp | 2 ++ tests/scheduler_tests.cpp | 2 ++ tests/tests.cpp | 2 ++ tests/utils_tests.cpp | 2 ++ tests/vmap_tests.cpp | 2 ++ 211 files changed, 449 insertions(+), 1 deletion(-) create mode 100644 ACKNOWLEDGMENTS.md diff --git a/ACKNOWLEDGMENTS.md b/ACKNOWLEDGMENTS.md new file mode 100644 index 00000000..93c6eb21 --- /dev/null +++ b/ACKNOWLEDGMENTS.md @@ -0,0 +1,30 @@ +MLX leverages several third-party software, listed here together with +their license copied verbatim. + +## PocketFFT + +Copyright (C) 2010-2018 Max-Planck-Society +All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, +are permitted provided that the following conditions are met: + +* Redistributions of source code must retain the above copyright notice, this + list of conditions and the following disclaimer. +* Redistributions in binary form must reproduce the above copyright notice, this + list of conditions and the following disclaimer in the documentation and/or + other materials provided with the distribution. +* Neither the name of the copyright holder nor the names of its contributors may + be used to endorse or promote products derived from this software without + specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED +WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE +DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR +ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES +(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; +LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. diff --git a/LICENSE b/LICENSE index bb7f0310..98ff47b9 100644 --- a/LICENSE +++ b/LICENSE @@ -1,6 +1,6 @@ MIT License -Copyright (c) 2023 ml-explore +Copyright © 2023 Apple Inc. Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal diff --git a/benchmarks/cpp/autograd.cpp b/benchmarks/cpp/autograd.cpp index fee3fc61..3beaa04b 100644 --- a/benchmarks/cpp/autograd.cpp +++ b/benchmarks/cpp/autograd.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/mlx.h" diff --git a/benchmarks/cpp/compare_devices.cpp b/benchmarks/cpp/compare_devices.cpp index 9e7c696e..eecbc9c4 100644 --- a/benchmarks/cpp/compare_devices.cpp +++ b/benchmarks/cpp/compare_devices.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/mlx.h" #include "time_utils.h" diff --git a/benchmarks/cpp/irregular_strides.cpp b/benchmarks/cpp/irregular_strides.cpp index e6a57099..50d4c5b7 100644 --- a/benchmarks/cpp/irregular_strides.cpp +++ b/benchmarks/cpp/irregular_strides.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/benchmarks/cpp/single_ops.cpp b/benchmarks/cpp/single_ops.cpp index aa0fa2b3..e8bc3a83 100644 --- a/benchmarks/cpp/single_ops.cpp +++ b/benchmarks/cpp/single_ops.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/mlx.h" #include "time_utils.h" diff --git a/benchmarks/cpp/time_utils.h b/benchmarks/cpp/time_utils.h index 9a4a0778..780f1867 100644 --- a/benchmarks/cpp/time_utils.h +++ b/benchmarks/cpp/time_utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/benchmarks/numpy/single_ops.py b/benchmarks/numpy/single_ops.py index e359d3ec..2631c50f 100644 --- a/benchmarks/numpy/single_ops.py +++ b/benchmarks/numpy/single_ops.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import numpy as np from time_utils import time_fn diff --git a/benchmarks/numpy/time_utils.py b/benchmarks/numpy/time_utils.py index 3741fba1..163f007d 100644 --- a/benchmarks/numpy/time_utils.py +++ b/benchmarks/numpy/time_utils.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import time diff --git a/benchmarks/python/batch_matmul_bench.py b/benchmarks/python/batch_matmul_bench.py index 11f0734e..84bf95fb 100644 --- a/benchmarks/python/batch_matmul_bench.py +++ b/benchmarks/python/batch_matmul_bench.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import mlx.core as mx diff --git a/benchmarks/python/blas/bench_gemm.py b/benchmarks/python/blas/bench_gemm.py index 27cd0793..608a7cb9 100644 --- a/benchmarks/python/blas/bench_gemm.py +++ b/benchmarks/python/blas/bench_gemm.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import numpy as np import argparse import mlx.core as mx diff --git a/benchmarks/python/blas/bench_gemv.py b/benchmarks/python/blas/bench_gemv.py index e95e48fa..41ac4ec3 100644 --- a/benchmarks/python/blas/bench_gemv.py +++ b/benchmarks/python/blas/bench_gemv.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import matplotlib.pyplot as plt import numpy as np import argparse diff --git a/benchmarks/python/comparative/bench_mlx.py b/benchmarks/python/comparative/bench_mlx.py index a8977427..5f2c3619 100644 --- a/benchmarks/python/comparative/bench_mlx.py +++ b/benchmarks/python/comparative/bench_mlx.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import math import os diff --git a/benchmarks/python/comparative/bench_torch.py b/benchmarks/python/comparative/bench_torch.py index 444ae6ce..f275b9da 100644 --- a/benchmarks/python/comparative/bench_torch.py +++ b/benchmarks/python/comparative/bench_torch.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import os import time diff --git a/benchmarks/python/comparative/compare.py b/benchmarks/python/comparative/compare.py index 00b9e407..f357165f 100644 --- a/benchmarks/python/comparative/compare.py +++ b/benchmarks/python/comparative/compare.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + #!/usr/bin/env python import argparse diff --git a/benchmarks/python/llama_jax_bench.py b/benchmarks/python/llama_jax_bench.py index e7539eca..9f647426 100644 --- a/benchmarks/python/llama_jax_bench.py +++ b/benchmarks/python/llama_jax_bench.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import time diff --git a/benchmarks/python/llama_mlx_bench.py b/benchmarks/python/llama_mlx_bench.py index f1cea473..b26ec24b 100644 --- a/benchmarks/python/llama_mlx_bench.py +++ b/benchmarks/python/llama_mlx_bench.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import time diff --git a/benchmarks/python/llama_torch_bench.py b/benchmarks/python/llama_torch_bench.py index 25b24484..72329f37 100644 --- a/benchmarks/python/llama_torch_bench.py +++ b/benchmarks/python/llama_torch_bench.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import time diff --git a/benchmarks/python/single_ops.py b/benchmarks/python/single_ops.py index f2977175..631eef4e 100644 --- a/benchmarks/python/single_ops.py +++ b/benchmarks/python/single_ops.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import mlx.core as mx diff --git a/benchmarks/python/time_utils.py b/benchmarks/python/time_utils.py index e067c809..73266cb7 100644 --- a/benchmarks/python/time_utils.py +++ b/benchmarks/python/time_utils.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import time import mlx.core as mx diff --git a/docs/src/conf.py b/docs/src/conf.py index 9018c231..5d3be4e5 100644 --- a/docs/src/conf.py +++ b/docs/src/conf.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + # -*- coding: utf-8 -*- import os diff --git a/examples/cpp/linear_regression.cpp b/examples/cpp/linear_regression.cpp index e6897b7a..f921da15 100644 --- a/examples/cpp/linear_regression.cpp +++ b/examples/cpp/linear_regression.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/examples/cpp/logistic_regression.cpp b/examples/cpp/logistic_regression.cpp index c34ae7fe..7f13f35b 100644 --- a/examples/cpp/logistic_regression.cpp +++ b/examples/cpp/logistic_regression.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/examples/cpp/timer.h b/examples/cpp/timer.h index 5a270f00..cea85ee8 100644 --- a/examples/cpp/timer.h +++ b/examples/cpp/timer.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/examples/cpp/tutorial.cpp b/examples/cpp/tutorial.cpp index 8426878e..5dc0e047 100644 --- a/examples/cpp/tutorial.cpp +++ b/examples/cpp/tutorial.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/examples/extensions/axpby/axpby.cpp b/examples/extensions/axpby/axpby.cpp index 1e6a995b..56a09f34 100644 --- a/examples/extensions/axpby/axpby.cpp +++ b/examples/extensions/axpby/axpby.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/examples/extensions/axpby/axpby.h b/examples/extensions/axpby/axpby.h index 6abc48a9..8c25980f 100644 --- a/examples/extensions/axpby/axpby.h +++ b/examples/extensions/axpby/axpby.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/ops.h" diff --git a/examples/extensions/axpby/axpby.metal b/examples/extensions/axpby/axpby.metal index fa284957..64980578 100644 --- a/examples/extensions/axpby/axpby.metal +++ b/examples/extensions/axpby/axpby.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/kernels/bf16.h" diff --git a/examples/extensions/bindings.cpp b/examples/extensions/bindings.cpp index d2eeca4a..661ddcba 100644 --- a/examples/extensions/bindings.cpp +++ b/examples/extensions/bindings.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/examples/extensions/mlx_sample_extensions/__init__.py b/examples/extensions/mlx_sample_extensions/__init__.py index 1b17da24..c194d099 100644 --- a/examples/extensions/mlx_sample_extensions/__init__.py +++ b/examples/extensions/mlx_sample_extensions/__init__.py @@ -1,2 +1,4 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx from .mlx_sample_extensions import * diff --git a/examples/extensions/setup.py b/examples/extensions/setup.py index c3817b04..a9ceb39d 100644 --- a/examples/extensions/setup.py +++ b/examples/extensions/setup.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from mlx import extension from setuptools import setup diff --git a/examples/python/linear_regression.py b/examples/python/linear_regression.py index f1837a64..47b0253d 100644 --- a/examples/python/linear_regression.py +++ b/examples/python/linear_regression.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx import time diff --git a/examples/python/logistic_regression.py b/examples/python/logistic_regression.py index 928ae34a..8df1fd99 100644 --- a/examples/python/logistic_regression.py +++ b/examples/python/logistic_regression.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx import time diff --git a/mlx/allocator.cpp b/mlx/allocator.cpp index 1d8cddd2..b591caee 100644 --- a/mlx/allocator.cpp +++ b/mlx/allocator.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/allocator.h b/mlx/allocator.h index 29421d47..2c3adadf 100644 --- a/mlx/allocator.h +++ b/mlx/allocator.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/array.cpp b/mlx/array.cpp index 20ea2e40..0a7b52a9 100644 --- a/mlx/array.cpp +++ b/mlx/array.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/array.h" diff --git a/mlx/array.h b/mlx/array.h index c0f02ee4..6f26c331 100644 --- a/mlx/array.h +++ b/mlx/array.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include #include diff --git a/mlx/backend/accelerate/conv.cpp b/mlx/backend/accelerate/conv.cpp index 432e60c5..22ddd907 100644 --- a/mlx/backend/accelerate/conv.cpp +++ b/mlx/backend/accelerate/conv.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/accelerate/matmul.cpp b/mlx/backend/accelerate/matmul.cpp index 5e8014c6..5a38e312 100644 --- a/mlx/backend/accelerate/matmul.cpp +++ b/mlx/backend/accelerate/matmul.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/accelerate/primitives.cpp b/mlx/backend/accelerate/primitives.cpp index 1a2b7b3e..4cde5bba 100644 --- a/mlx/backend/accelerate/primitives.cpp +++ b/mlx/backend/accelerate/primitives.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/accelerate/reduce.cpp b/mlx/backend/accelerate/reduce.cpp index 19de62f5..db2b8eba 100644 --- a/mlx/backend/accelerate/reduce.cpp +++ b/mlx/backend/accelerate/reduce.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/accelerate/softmax.cpp b/mlx/backend/accelerate/softmax.cpp index 31cdbd2d..fcd8fbe5 100644 --- a/mlx/backend/accelerate/softmax.cpp +++ b/mlx/backend/accelerate/softmax.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/accelerate/utils.h b/mlx/backend/accelerate/utils.h index 7c2f1ae6..1dbabe30 100644 --- a/mlx/backend/accelerate/utils.h +++ b/mlx/backend/accelerate/utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/common/arange.h b/mlx/backend/common/arange.h index 05d87488..5c155dc0 100644 --- a/mlx/backend/common/arange.h +++ b/mlx/backend/common/arange.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/allocator.h" diff --git a/mlx/backend/common/arg_reduce.cpp b/mlx/backend/common/arg_reduce.cpp index 221df856..f4a67259 100644 --- a/mlx/backend/common/arg_reduce.cpp +++ b/mlx/backend/common/arg_reduce.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/primitives.h" diff --git a/mlx/backend/common/binary.cpp b/mlx/backend/common/binary.cpp index 3e4166a7..38f0461c 100644 --- a/mlx/backend/common/binary.cpp +++ b/mlx/backend/common/binary.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/common/binary.h b/mlx/backend/common/binary.h index 744ab520..4b063174 100644 --- a/mlx/backend/common/binary.h +++ b/mlx/backend/common/binary.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/allocator.h" diff --git a/mlx/backend/common/conv.cpp b/mlx/backend/common/conv.cpp index 863aa941..f46a68d6 100644 --- a/mlx/backend/common/conv.cpp +++ b/mlx/backend/common/conv.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #ifdef ACCELERATE_NEW_LAPACK diff --git a/mlx/backend/common/copy.cpp b/mlx/backend/common/copy.cpp index 36d79ac9..c8af4b54 100644 --- a/mlx/backend/common/copy.cpp +++ b/mlx/backend/common/copy.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/allocator.h" diff --git a/mlx/backend/common/copy.h b/mlx/backend/common/copy.h index dc46e91b..0affddec 100644 --- a/mlx/backend/common/copy.h +++ b/mlx/backend/common/copy.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/array.h" diff --git a/mlx/backend/common/default_primitives.cpp b/mlx/backend/common/default_primitives.cpp index 3ed9f275..a73853ed 100644 --- a/mlx/backend/common/default_primitives.cpp +++ b/mlx/backend/common/default_primitives.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/array.h" diff --git a/mlx/backend/common/erf.cpp b/mlx/backend/common/erf.cpp index 0b876a1b..83769d07 100644 --- a/mlx/backend/common/erf.cpp +++ b/mlx/backend/common/erf.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include namespace mlx::core { diff --git a/mlx/backend/common/erf.h b/mlx/backend/common/erf.h index d7698754..25a53d37 100644 --- a/mlx/backend/common/erf.h +++ b/mlx/backend/common/erf.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + namespace mlx::core { diff --git a/mlx/backend/common/fft.cpp b/mlx/backend/common/fft.cpp index 385d9cb7..e46fd7f9 100644 --- a/mlx/backend/common/fft.cpp +++ b/mlx/backend/common/fft.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/3rdparty/pocketfft.h" diff --git a/mlx/backend/common/indexing.cpp b/mlx/backend/common/indexing.cpp index 9572c1ca..64fc21ec 100644 --- a/mlx/backend/common/indexing.cpp +++ b/mlx/backend/common/indexing.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/common/load.cpp b/mlx/backend/common/load.cpp index 5f5f1732..e68ce7f6 100644 --- a/mlx/backend/common/load.cpp +++ b/mlx/backend/common/load.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/common/primitives.cpp b/mlx/backend/common/primitives.cpp index 86bc746c..4049af2e 100644 --- a/mlx/backend/common/primitives.cpp +++ b/mlx/backend/common/primitives.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/common/reduce.cpp b/mlx/backend/common/reduce.cpp index 90a466f7..e9d091e4 100644 --- a/mlx/backend/common/reduce.cpp +++ b/mlx/backend/common/reduce.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/common/reduce.h b/mlx/backend/common/reduce.h index 31b1716c..740f54a4 100644 --- a/mlx/backend/common/reduce.h +++ b/mlx/backend/common/reduce.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/backend/common/utils.h" diff --git a/mlx/backend/common/scan.cpp b/mlx/backend/common/scan.cpp index 89e09e37..9f994b90 100644 --- a/mlx/backend/common/scan.cpp +++ b/mlx/backend/common/scan.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/common/copy.h" diff --git a/mlx/backend/common/softmax.cpp b/mlx/backend/common/softmax.cpp index dabf8114..90874c72 100644 --- a/mlx/backend/common/softmax.cpp +++ b/mlx/backend/common/softmax.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/common/sort.cpp b/mlx/backend/common/sort.cpp index 8f029559..edeee065 100644 --- a/mlx/backend/common/sort.cpp +++ b/mlx/backend/common/sort.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/common/threefry.cpp b/mlx/backend/common/threefry.cpp index 391648ce..b4905acb 100644 --- a/mlx/backend/common/threefry.cpp +++ b/mlx/backend/common/threefry.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/backend/common/threefry.h" namespace mlx::core::random { diff --git a/mlx/backend/common/threefry.h b/mlx/backend/common/threefry.h index efcb5dcf..0fc485fc 100644 --- a/mlx/backend/common/threefry.h +++ b/mlx/backend/common/threefry.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/common/unary.h b/mlx/backend/common/unary.h index 271215b5..dbbaf699 100644 --- a/mlx/backend/common/unary.h +++ b/mlx/backend/common/unary.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/allocator.h" diff --git a/mlx/backend/common/utils.h b/mlx/backend/common/utils.h index 1cc7e9f5..e170f5df 100644 --- a/mlx/backend/common/utils.h +++ b/mlx/backend/common/utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/allocator.cpp b/mlx/backend/metal/allocator.cpp index 40ec85a1..a5569094 100644 --- a/mlx/backend/metal/allocator.cpp +++ b/mlx/backend/metal/allocator.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/backend/metal/allocator.h" #include "mlx/backend/metal/metal.h" diff --git a/mlx/backend/metal/allocator.h b/mlx/backend/metal/allocator.h index 92ede175..621b78a9 100644 --- a/mlx/backend/metal/allocator.h +++ b/mlx/backend/metal/allocator.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/conv.cpp b/mlx/backend/metal/conv.cpp index e1329717..e25599ca 100644 --- a/mlx/backend/metal/conv.cpp +++ b/mlx/backend/metal/conv.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/copy.cpp b/mlx/backend/metal/copy.cpp index aadb98e9..c14c1022 100644 --- a/mlx/backend/metal/copy.cpp +++ b/mlx/backend/metal/copy.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/copy.h" diff --git a/mlx/backend/metal/copy.h b/mlx/backend/metal/copy.h index 02d294a8..c08ae2cf 100644 --- a/mlx/backend/metal/copy.h +++ b/mlx/backend/metal/copy.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/backend/common/copy.h" diff --git a/mlx/backend/metal/device.cpp b/mlx/backend/metal/device.cpp index 6a133329..24251ca0 100644 --- a/mlx/backend/metal/device.cpp +++ b/mlx/backend/metal/device.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/device.h b/mlx/backend/metal/device.h index bad00157..62675d43 100644 --- a/mlx/backend/metal/device.h +++ b/mlx/backend/metal/device.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/fft.cpp b/mlx/backend/metal/fft.cpp index 0cf34e5d..c0844d37 100644 --- a/mlx/backend/metal/fft.cpp +++ b/mlx/backend/metal/fft.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/primitives.h" namespace mlx::core { diff --git a/mlx/backend/metal/indexing.cpp b/mlx/backend/metal/indexing.cpp index ea5b771f..eb9a8efb 100644 --- a/mlx/backend/metal/indexing.cpp +++ b/mlx/backend/metal/indexing.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/kernels/arange.metal b/mlx/backend/metal/kernels/arange.metal index f3afc24f..d7ba56aa 100644 --- a/mlx/backend/metal/kernels/arange.metal +++ b/mlx/backend/metal/kernels/arange.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/backend/metal/kernels/bf16.h" template diff --git a/mlx/backend/metal/kernels/arg_reduce.metal b/mlx/backend/metal/kernels/arg_reduce.metal index 8cd13afc..31bcbfa0 100644 --- a/mlx/backend/metal/kernels/arg_reduce.metal +++ b/mlx/backend/metal/kernels/arg_reduce.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/kernels/atomic.h b/mlx/backend/metal/kernels/atomic.h index 43c8cc3b..1c260d1b 100644 --- a/mlx/backend/metal/kernels/atomic.h +++ b/mlx/backend/metal/kernels/atomic.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/kernels/bf16.h b/mlx/backend/metal/kernels/bf16.h index 618aefd4..71a45e1c 100644 --- a/mlx/backend/metal/kernels/bf16.h +++ b/mlx/backend/metal/kernels/bf16.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/kernels/bf16_math.h b/mlx/backend/metal/kernels/bf16_math.h index 824ecc9e..f1fec433 100644 --- a/mlx/backend/metal/kernels/bf16_math.h +++ b/mlx/backend/metal/kernels/bf16_math.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/backend/metal/kernels/bf16.h" diff --git a/mlx/backend/metal/kernels/binary.metal b/mlx/backend/metal/kernels/binary.metal index 5baab0e9..c22bfde3 100644 --- a/mlx/backend/metal/kernels/binary.metal +++ b/mlx/backend/metal/kernels/binary.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/kernels/complex.h b/mlx/backend/metal/kernels/complex.h index b4be6d51..0bfa707f 100644 --- a/mlx/backend/metal/kernels/complex.h +++ b/mlx/backend/metal/kernels/complex.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/kernels/conv.metal b/mlx/backend/metal/kernels/conv.metal index 65329d19..6dbe7bc3 100644 --- a/mlx/backend/metal/kernels/conv.metal +++ b/mlx/backend/metal/kernels/conv.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/kernels/conv_params.h" diff --git a/mlx/backend/metal/kernels/conv_params.h b/mlx/backend/metal/kernels/conv_params.h index b3748ef4..b216bb97 100644 --- a/mlx/backend/metal/kernels/conv_params.h +++ b/mlx/backend/metal/kernels/conv_params.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once template diff --git a/mlx/backend/metal/kernels/copy.metal b/mlx/backend/metal/kernels/copy.metal index 07ae22ba..9808cc95 100644 --- a/mlx/backend/metal/kernels/copy.metal +++ b/mlx/backend/metal/kernels/copy.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/backend/metal/kernels/bf16.h" #include "mlx/backend/metal/kernels/utils.h" diff --git a/mlx/backend/metal/kernels/defines.h b/mlx/backend/metal/kernels/defines.h index 38102e98..bdd1419f 100644 --- a/mlx/backend/metal/kernels/defines.h +++ b/mlx/backend/metal/kernels/defines.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #ifdef __METAL__ diff --git a/mlx/backend/metal/kernels/erf.h b/mlx/backend/metal/kernels/erf.h index de9cc58c..0a370d30 100644 --- a/mlx/backend/metal/kernels/erf.h +++ b/mlx/backend/metal/kernels/erf.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/kernels/gemm.metal b/mlx/backend/metal/kernels/gemm.metal index 9f51e6ff..df150b50 100644 --- a/mlx/backend/metal/kernels/gemm.metal +++ b/mlx/backend/metal/kernels/gemm.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/backend/metal/kernels/bf16.h" #include "mlx/backend/metal/kernels/gemm/gemm.h" diff --git a/mlx/backend/metal/kernels/gemm/conv.h b/mlx/backend/metal/kernels/gemm/conv.h index faf550e3..2c4a7074 100644 --- a/mlx/backend/metal/kernels/gemm/conv.h +++ b/mlx/backend/metal/kernels/gemm/conv.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/kernels/gemm/gemm.h b/mlx/backend/metal/kernels/gemm/gemm.h index f6c7b583..f551947d 100644 --- a/mlx/backend/metal/kernels/gemm/gemm.h +++ b/mlx/backend/metal/kernels/gemm/gemm.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/kernels/gemv.metal b/mlx/backend/metal/kernels/gemv.metal index 9b484871..ea52ab57 100644 --- a/mlx/backend/metal/kernels/gemv.metal +++ b/mlx/backend/metal/kernels/gemv.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/kernels/indexing.metal b/mlx/backend/metal/kernels/indexing.metal index fc0d9ee7..daba39a4 100644 --- a/mlx/backend/metal/kernels/indexing.metal +++ b/mlx/backend/metal/kernels/indexing.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/kernels/random.metal b/mlx/backend/metal/kernels/random.metal index d8e4c420..cc9c5fe3 100644 --- a/mlx/backend/metal/kernels/random.metal +++ b/mlx/backend/metal/kernels/random.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/backend/metal/kernels/utils.h" static constexpr constant uint32_t rotations[2][4] = { diff --git a/mlx/backend/metal/kernels/reduce.h b/mlx/backend/metal/kernels/reduce.h index d670f245..1d2b971b 100644 --- a/mlx/backend/metal/kernels/reduce.h +++ b/mlx/backend/metal/kernels/reduce.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/kernels/reduce.metal b/mlx/backend/metal/kernels/reduce.metal index f6951490..25bf1ee1 100644 --- a/mlx/backend/metal/kernels/reduce.metal +++ b/mlx/backend/metal/kernels/reduce.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/kernels/scan.metal b/mlx/backend/metal/kernels/scan.metal index ba2368e2..e692449c 100644 --- a/mlx/backend/metal/kernels/scan.metal +++ b/mlx/backend/metal/kernels/scan.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/kernels/softmax.metal b/mlx/backend/metal/kernels/softmax.metal index 07a49bc0..0877de07 100644 --- a/mlx/backend/metal/kernels/softmax.metal +++ b/mlx/backend/metal/kernels/softmax.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/kernels/sort.metal b/mlx/backend/metal/kernels/sort.metal index adafe6e1..7e2c0442 100644 --- a/mlx/backend/metal/kernels/sort.metal +++ b/mlx/backend/metal/kernels/sort.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/kernels/bf16.h" diff --git a/mlx/backend/metal/kernels/unary.metal b/mlx/backend/metal/kernels/unary.metal index 213e440d..bfda9938 100644 --- a/mlx/backend/metal/kernels/unary.metal +++ b/mlx/backend/metal/kernels/unary.metal @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/kernels/utils.h b/mlx/backend/metal/kernels/utils.h index 22b4a271..72cdd8b2 100644 --- a/mlx/backend/metal/kernels/utils.h +++ b/mlx/backend/metal/kernels/utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/matmul.cpp b/mlx/backend/metal/matmul.cpp index d6fdc34e..22472c63 100644 --- a/mlx/backend/metal/matmul.cpp +++ b/mlx/backend/metal/matmul.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/matmul.h b/mlx/backend/metal/matmul.h index 34e86733..78a0e7f2 100644 --- a/mlx/backend/metal/matmul.h +++ b/mlx/backend/metal/matmul.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/metal.cpp b/mlx/backend/metal/metal.cpp index d936ff3e..f63ad55a 100644 --- a/mlx/backend/metal/metal.cpp +++ b/mlx/backend/metal/metal.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/metal.h b/mlx/backend/metal/metal.h index 8195080f..f1f7ede4 100644 --- a/mlx/backend/metal/metal.h +++ b/mlx/backend/metal/metal.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/mps/gemm.h b/mlx/backend/metal/mps/gemm.h index 62016446..c93df68d 100644 --- a/mlx/backend/metal/mps/gemm.h +++ b/mlx/backend/metal/mps/gemm.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/backend/metal/primitives.cpp b/mlx/backend/metal/primitives.cpp index 48ceb054..7860d7de 100644 --- a/mlx/backend/metal/primitives.cpp +++ b/mlx/backend/metal/primitives.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/reduce.cpp b/mlx/backend/metal/reduce.cpp index 84a368c5..532f1835 100644 --- a/mlx/backend/metal/reduce.cpp +++ b/mlx/backend/metal/reduce.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/backend/metal/scan.cpp b/mlx/backend/metal/scan.cpp index 9c91f89e..92f1007e 100644 --- a/mlx/backend/metal/scan.cpp +++ b/mlx/backend/metal/scan.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/backend/metal/softmax.cpp b/mlx/backend/metal/softmax.cpp index 7f7b51b9..33ec8014 100644 --- a/mlx/backend/metal/softmax.cpp +++ b/mlx/backend/metal/softmax.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/copy.h" diff --git a/mlx/backend/metal/sort.cpp b/mlx/backend/metal/sort.cpp index 8cffdd4e..befbf2d8 100644 --- a/mlx/backend/metal/sort.cpp +++ b/mlx/backend/metal/sort.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/copy.h" diff --git a/mlx/backend/metal/utils.h b/mlx/backend/metal/utils.h index 0165761e..6fa08e42 100644 --- a/mlx/backend/metal/utils.h +++ b/mlx/backend/metal/utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/array.h" diff --git a/mlx/backend/no_metal/allocator.cpp b/mlx/backend/no_metal/allocator.cpp index 718f3509..dcb67d93 100644 --- a/mlx/backend/no_metal/allocator.cpp +++ b/mlx/backend/no_metal/allocator.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/allocator.h" diff --git a/mlx/backend/no_metal/metal.cpp b/mlx/backend/no_metal/metal.cpp index ec451888..accfc4c8 100644 --- a/mlx/backend/no_metal/metal.cpp +++ b/mlx/backend/no_metal/metal.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/metal.h" diff --git a/mlx/backend/no_metal/primitives.cpp b/mlx/backend/no_metal/primitives.cpp index 7a42e3ff..3c327c67 100644 --- a/mlx/backend/no_metal/primitives.cpp +++ b/mlx/backend/no_metal/primitives.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/primitives.h" diff --git a/mlx/device.cpp b/mlx/device.cpp index 352b0284..e635782e 100644 --- a/mlx/device.cpp +++ b/mlx/device.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/device.h" #include "mlx/backend/metal/metal.h" diff --git a/mlx/device.h b/mlx/device.h index 66035ddc..e11edf79 100644 --- a/mlx/device.h +++ b/mlx/device.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once namespace mlx::core { diff --git a/mlx/dtype.cpp b/mlx/dtype.cpp index 23749e9a..99b75a02 100644 --- a/mlx/dtype.cpp +++ b/mlx/dtype.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/dtype.h b/mlx/dtype.h index 6a5f1179..cdb8aa59 100644 --- a/mlx/dtype.h +++ b/mlx/dtype.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/fft.cpp b/mlx/fft.cpp index 96dc76ba..6cb33048 100644 --- a/mlx/fft.cpp +++ b/mlx/fft.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/fft.h b/mlx/fft.h index 62a34f4e..dbcc777f 100644 --- a/mlx/fft.h +++ b/mlx/fft.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/graph_utils.cpp b/mlx/graph_utils.cpp index 59bd3736..7c83fdea 100644 --- a/mlx/graph_utils.cpp +++ b/mlx/graph_utils.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/graph_utils.h b/mlx/graph_utils.h index e0696e5e..3bd373be 100644 --- a/mlx/graph_utils.h +++ b/mlx/graph_utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/array.h" diff --git a/mlx/load.cpp b/mlx/load.cpp index 1d3b571e..8106448a 100644 --- a/mlx/load.cpp +++ b/mlx/load.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/load.h b/mlx/load.h index 8b433ac2..1d193392 100644 --- a/mlx/load.h +++ b/mlx/load.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/mlx.h b/mlx/mlx.h index 68b6c182..102d2dde 100644 --- a/mlx/mlx.h +++ b/mlx/mlx.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/array.h" diff --git a/mlx/ops.cpp b/mlx/ops.cpp index c851a44a..c279b76f 100644 --- a/mlx/ops.cpp +++ b/mlx/ops.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/ops.h b/mlx/ops.h index 50666fa2..552226b8 100644 --- a/mlx/ops.h +++ b/mlx/ops.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/primitives.cpp b/mlx/primitives.cpp index 8a2ade91..4c07e14e 100644 --- a/mlx/primitives.cpp +++ b/mlx/primitives.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/primitives.h b/mlx/primitives.h index 44159f86..3497bc26 100644 --- a/mlx/primitives.h +++ b/mlx/primitives.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "array.h" diff --git a/mlx/random.cpp b/mlx/random.cpp index 41ce2b5a..e976c2b8 100644 --- a/mlx/random.cpp +++ b/mlx/random.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/random.h b/mlx/random.h index 25dc9f92..e684464b 100644 --- a/mlx/random.h +++ b/mlx/random.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/scheduler.cpp b/mlx/scheduler.cpp index cb2842a5..af66af06 100644 --- a/mlx/scheduler.cpp +++ b/mlx/scheduler.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "mlx/scheduler.h" #include "mlx/backend/metal/metal.h" diff --git a/mlx/scheduler.h b/mlx/scheduler.h index ba615d37..6506b20a 100644 --- a/mlx/scheduler.h +++ b/mlx/scheduler.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/stream.h b/mlx/stream.h index 000185f0..d7b4268f 100644 --- a/mlx/stream.h +++ b/mlx/stream.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "mlx/device.h" diff --git a/mlx/transforms.cpp b/mlx/transforms.cpp index 89e0c42a..174d5e37 100644 --- a/mlx/transforms.cpp +++ b/mlx/transforms.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/mlx/transforms.h b/mlx/transforms.h index 88296521..ff47638b 100644 --- a/mlx/transforms.h +++ b/mlx/transforms.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "array.h" diff --git a/mlx/transforms_impl.h b/mlx/transforms_impl.h index e97d41d7..0fa8f06a 100644 --- a/mlx/transforms_impl.h +++ b/mlx/transforms_impl.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + namespace mlx::core::detail { diff --git a/mlx/types/bf16.h b/mlx/types/bf16.h index 102aea91..59519417 100644 --- a/mlx/types/bf16.h +++ b/mlx/types/bf16.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/types/complex.h b/mlx/types/complex.h index b7533efb..55cbe447 100644 --- a/mlx/types/complex.h +++ b/mlx/types/complex.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include #include "mlx/types/half_types.h" diff --git a/mlx/types/fp16.h b/mlx/types/fp16.h index e4f94c99..58e1bc08 100644 --- a/mlx/types/fp16.h +++ b/mlx/types/fp16.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/mlx/types/half_types.h b/mlx/types/half_types.h index 47ea3d08..43027956 100644 --- a/mlx/types/half_types.h +++ b/mlx/types/half_types.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #ifdef __ARM_FEATURE_FP16_SCALAR_ARITHMETIC diff --git a/mlx/utils.cpp b/mlx/utils.cpp index 027e104e..5f38cd06 100644 --- a/mlx/utils.cpp +++ b/mlx/utils.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/mlx/utils.h b/mlx/utils.h index b004ae8b..f21e55fc 100644 --- a/mlx/utils.h +++ b/mlx/utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include "array.h" diff --git a/python/mlx/_reprlib_fix.py b/python/mlx/_reprlib_fix.py index 51111e5d..02bf661d 100644 --- a/python/mlx/_reprlib_fix.py +++ b/python/mlx/_reprlib_fix.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import array import reprlib diff --git a/python/mlx/extension.py b/python/mlx/extension.py index 8dd7a6a7..2924ef0c 100644 --- a/python/mlx/extension.py +++ b/python/mlx/extension.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import os import re import subprocess diff --git a/python/mlx/nn/__init__.py b/python/mlx/nn/__init__.py index 4991254f..2af4f3a5 100644 --- a/python/mlx/nn/__init__.py +++ b/python/mlx/nn/__init__.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from mlx.nn.layers import * from mlx.nn import losses from mlx.nn.utils import value_and_grad diff --git a/python/mlx/nn/layers/__init__.py b/python/mlx/nn/layers/__init__.py index 0436441a..30a18504 100644 --- a/python/mlx/nn/layers/__init__.py +++ b/python/mlx/nn/layers/__init__.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from mlx.nn.layers.base import Module from mlx.nn.layers.activations import ( GELU, diff --git a/python/mlx/nn/layers/activations.py b/python/mlx/nn/layers/activations.py index ef1cde03..dbe5f4ad 100644 --- a/python/mlx/nn/layers/activations.py +++ b/python/mlx/nn/layers/activations.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import mlx.core as mx diff --git a/python/mlx/nn/layers/base.py b/python/mlx/nn/layers/base.py index 3a3cd3d2..991b1381 100644 --- a/python/mlx/nn/layers/base.py +++ b/python/mlx/nn/layers/base.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import textwrap from typing import Any, Callable, List, Union, Optional diff --git a/python/mlx/nn/layers/containers.py b/python/mlx/nn/layers/containers.py index 8403be06..c57fbb99 100644 --- a/python/mlx/nn/layers/containers.py +++ b/python/mlx/nn/layers/containers.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from mlx.nn.layers.base import Module diff --git a/python/mlx/nn/layers/convolution.py b/python/mlx/nn/layers/convolution.py index 1e7ac1c3..c9b6ce36 100644 --- a/python/mlx/nn/layers/convolution.py +++ b/python/mlx/nn/layers/convolution.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math from typing import Union diff --git a/python/mlx/nn/layers/dropout.py b/python/mlx/nn/layers/dropout.py index e6e88938..3193cdbd 100644 --- a/python/mlx/nn/layers/dropout.py +++ b/python/mlx/nn/layers/dropout.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx from mlx.nn.layers.base import Module diff --git a/python/mlx/nn/layers/embedding.py b/python/mlx/nn/layers/embedding.py index d6e9b976..c62b1206 100644 --- a/python/mlx/nn/layers/embedding.py +++ b/python/mlx/nn/layers/embedding.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import mlx.core as mx diff --git a/python/mlx/nn/layers/linear.py b/python/mlx/nn/layers/linear.py index aa48688d..d4a50338 100644 --- a/python/mlx/nn/layers/linear.py +++ b/python/mlx/nn/layers/linear.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import mlx.core as mx diff --git a/python/mlx/nn/layers/normalization.py b/python/mlx/nn/layers/normalization.py index 6d0dd4b3..348a8c54 100644 --- a/python/mlx/nn/layers/normalization.py +++ b/python/mlx/nn/layers/normalization.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx from mlx.nn.layers.base import Module diff --git a/python/mlx/nn/layers/positional_encoding.py b/python/mlx/nn/layers/positional_encoding.py index 9ccc2b69..5c3171fd 100644 --- a/python/mlx/nn/layers/positional_encoding.py +++ b/python/mlx/nn/layers/positional_encoding.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math from typing import Optional diff --git a/python/mlx/nn/layers/transformer.py b/python/mlx/nn/layers/transformer.py index 93359a4c..d2c224cb 100644 --- a/python/mlx/nn/layers/transformer.py +++ b/python/mlx/nn/layers/transformer.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math from typing import Optional diff --git a/python/mlx/nn/losses.py b/python/mlx/nn/losses.py index 800242c7..dfa806c3 100644 --- a/python/mlx/nn/losses.py +++ b/python/mlx/nn/losses.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx diff --git a/python/mlx/nn/utils.py b/python/mlx/nn/utils.py index c1023619..8ab80cb0 100644 --- a/python/mlx/nn/utils.py +++ b/python/mlx/nn/utils.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from typing import Callable import mlx.core as mx diff --git a/python/mlx/optimizers.py b/python/mlx/optimizers.py index 63ccf03d..4fb48935 100644 --- a/python/mlx/optimizers.py +++ b/python/mlx/optimizers.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math from typing import List diff --git a/python/mlx/utils.py b/python/mlx/utils.py index 47842d83..d70b8ce5 100644 --- a/python/mlx/utils.py +++ b/python/mlx/utils.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + def tree_map(fn, tree, *rest): """Applies ``fn`` to the leaves of the python tree ``tree`` and returns a new collection with the results. diff --git a/python/src/array.cpp b/python/src/array.cpp index 5e34bfdb..ea580c78 100644 --- a/python/src/array.cpp +++ b/python/src/array.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/python/src/device.cpp b/python/src/device.cpp index 45ae44ae..50438c13 100644 --- a/python/src/device.cpp +++ b/python/src/device.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/python/src/fft.cpp b/python/src/fft.cpp index d20e2b2f..42ad3763 100644 --- a/python/src/fft.cpp +++ b/python/src/fft.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/python/src/indexing.cpp b/python/src/indexing.cpp index 968a5374..788e2fd5 100644 --- a/python/src/indexing.cpp +++ b/python/src/indexing.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/python/src/indexing.h b/python/src/indexing.h index cb5d3106..0eaaac93 100644 --- a/python/src/indexing.h +++ b/python/src/indexing.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/python/src/load.cpp b/python/src/load.cpp index d3f1f925..b0327e12 100644 --- a/python/src/load.cpp +++ b/python/src/load.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/python/src/load.h b/python/src/load.h index f7f1a414..bbabef2c 100644 --- a/python/src/load.h +++ b/python/src/load.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include diff --git a/python/src/metal.cpp b/python/src/metal.cpp index 75e0437b..41b90d97 100644 --- a/python/src/metal.cpp +++ b/python/src/metal.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "mlx/backend/metal/metal.h" diff --git a/python/src/mlx.cpp b/python/src/mlx.cpp index 0e9aeba6..4a1d24ae 100644 --- a/python/src/mlx.cpp +++ b/python/src/mlx.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #define STRINGIFY(x) #x diff --git a/python/src/ops.cpp b/python/src/ops.cpp index ca33cabd..e32c01bc 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/python/src/random.cpp b/python/src/random.cpp index f38fc43d..943370db 100644 --- a/python/src/random.cpp +++ b/python/src/random.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/python/src/stream.cpp b/python/src/stream.cpp index f7a73463..7b1b2f55 100644 --- a/python/src/stream.cpp +++ b/python/src/stream.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/python/src/transforms.cpp b/python/src/transforms.cpp index b257ab40..a592b445 100644 --- a/python/src/transforms.cpp +++ b/python/src/transforms.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/python/src/utils.h b/python/src/utils.h index bc06fb50..142f9a98 100644 --- a/python/src/utils.h +++ b/python/src/utils.h @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #pragma once #include #include diff --git a/python/tests/mlx_tests.py b/python/tests/mlx_tests.py index 6c19f6eb..7bbadc9c 100644 --- a/python/tests/mlx_tests.py +++ b/python/tests/mlx_tests.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import os import unittest diff --git a/python/tests/test_array.py b/python/tests/test_array.py index e7edb6c2..16579742 100644 --- a/python/tests/test_array.py +++ b/python/tests/test_array.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import operator import unittest from itertools import permutations diff --git a/python/tests/test_autograd.py b/python/tests/test_autograd.py index 96e22dd6..5cead2f3 100644 --- a/python/tests/test_autograd.py +++ b/python/tests/test_autograd.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/python/tests/test_bf16.py b/python/tests/test_bf16.py index e73937d7..27a4e562 100644 --- a/python/tests/test_bf16.py +++ b/python/tests/test_bf16.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest from itertools import permutations diff --git a/python/tests/test_blas.py b/python/tests/test_blas.py index 3a81e10c..4b310ed3 100644 --- a/python/tests/test_blas.py +++ b/python/tests/test_blas.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest from itertools import permutations diff --git a/python/tests/test_conv.py b/python/tests/test_conv.py index c4df6035..13b51e71 100644 --- a/python/tests/test_conv.py +++ b/python/tests/test_conv.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest from itertools import permutations diff --git a/python/tests/test_device.py b/python/tests/test_device.py index b3abec01..fff368b9 100644 --- a/python/tests/test_device.py +++ b/python/tests/test_device.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/python/tests/test_eval.py b/python/tests/test_eval.py index e4d48a87..88149729 100644 --- a/python/tests/test_eval.py +++ b/python/tests/test_eval.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from functools import partial import unittest diff --git a/python/tests/test_fft.py b/python/tests/test_fft.py index 6af90e1c..ce273bb2 100644 --- a/python/tests/test_fft.py +++ b/python/tests/test_fft.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import itertools diff --git a/python/tests/test_load.py b/python/tests/test_load.py index ebaddf98..6f345fa9 100644 --- a/python/tests/test_load.py +++ b/python/tests/test_load.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import os import mlx.core as mx diff --git a/python/tests/test_nn.py b/python/tests/test_nn.py index cc664cbb..77eb6bf7 100644 --- a/python/tests/test_nn.py +++ b/python/tests/test_nn.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/python/tests/test_ops.py b/python/tests/test_ops.py index 9885a7e7..7def06c2 100644 --- a/python/tests/test_ops.py +++ b/python/tests/test_ops.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest from itertools import permutations diff --git a/python/tests/test_optimizers.py b/python/tests/test_optimizers.py index 7cfc523f..3d4f4518 100644 --- a/python/tests/test_optimizers.py +++ b/python/tests/test_optimizers.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/python/tests/test_random.py b/python/tests/test_random.py index 209d6d9e..e850e423 100644 --- a/python/tests/test_random.py +++ b/python/tests/test_random.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/python/tests/test_reduce.py b/python/tests/test_reduce.py index b75f8a0b..da620732 100644 --- a/python/tests/test_reduce.py +++ b/python/tests/test_reduce.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest from itertools import permutations, combinations diff --git a/python/tests/test_tree.py b/python/tests/test_tree.py index fe6d8cdb..f5003c4a 100644 --- a/python/tests/test_tree.py +++ b/python/tests/test_tree.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/python/tests/test_vmap.py b/python/tests/test_vmap.py index 151fbc3c..50f7e6ef 100644 --- a/python/tests/test_vmap.py +++ b/python/tests/test_vmap.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/setup.py b/setup.py index c4d3569c..c5e6c1de 100644 --- a/setup.py +++ b/setup.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import os import re import subprocess diff --git a/tests/allocator_tests.cpp b/tests/allocator_tests.cpp index 70199eb4..0c1f3000 100644 --- a/tests/allocator_tests.cpp +++ b/tests/allocator_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "doctest/doctest.h" diff --git a/tests/arg_reduce_tests.cpp b/tests/arg_reduce_tests.cpp index 2bfd8968..7e3750e7 100644 --- a/tests/arg_reduce_tests.cpp +++ b/tests/arg_reduce_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "doctest/doctest.h" diff --git a/tests/array_tests.cpp b/tests/array_tests.cpp index c435b1db..334b8611 100644 --- a/tests/array_tests.cpp +++ b/tests/array_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "doctest/doctest.h" diff --git a/tests/autograd_tests.cpp b/tests/autograd_tests.cpp index 7c7245cd..85dad916 100644 --- a/tests/autograd_tests.cpp +++ b/tests/autograd_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/tests/blas_tests.cpp b/tests/blas_tests.cpp index 256a876a..f700e8c8 100644 --- a/tests/blas_tests.cpp +++ b/tests/blas_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "doctest/doctest.h" diff --git a/tests/creations_tests.cpp b/tests/creations_tests.cpp index 8d03ff1f..9d5f787c 100644 --- a/tests/creations_tests.cpp +++ b/tests/creations_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" #include "mlx/mlx.h" diff --git a/tests/device_tests.cpp b/tests/device_tests.cpp index dedf6457..87d55d64 100644 --- a/tests/device_tests.cpp +++ b/tests/device_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" #include diff --git a/tests/eval_tests.cpp b/tests/eval_tests.cpp index 88af76d5..dc2e96bb 100644 --- a/tests/eval_tests.cpp +++ b/tests/eval_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" #include "mlx/mlx.h" diff --git a/tests/fft_tests.cpp b/tests/fft_tests.cpp index c6940751..5d960ef1 100644 --- a/tests/fft_tests.cpp +++ b/tests/fft_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" #include "mlx/mlx.h" diff --git a/tests/graph_optimize_tests.cpp b/tests/graph_optimize_tests.cpp index 669bb08e..8c2da162 100644 --- a/tests/graph_optimize_tests.cpp +++ b/tests/graph_optimize_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" #include "mlx/mlx.h" diff --git a/tests/load_tests.cpp b/tests/load_tests.cpp index 45f36e47..f2489ca7 100644 --- a/tests/load_tests.cpp +++ b/tests/load_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include #include diff --git a/tests/metal_tests.cpp b/tests/metal_tests.cpp index 76f62dce..adeeb742 100644 --- a/tests/metal_tests.cpp +++ b/tests/metal_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "doctest/doctest.h" diff --git a/tests/ops_tests.cpp b/tests/ops_tests.cpp index a9a74b82..5dcf8658 100644 --- a/tests/ops_tests.cpp +++ b/tests/ops_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include diff --git a/tests/random_tests.cpp b/tests/random_tests.cpp index 92ed802d..c878205e 100644 --- a/tests/random_tests.cpp +++ b/tests/random_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include #include "doctest/doctest.h" diff --git a/tests/scheduler_tests.cpp b/tests/scheduler_tests.cpp index 781b2aa7..d30d01cf 100644 --- a/tests/scheduler_tests.cpp +++ b/tests/scheduler_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" #include "mlx/mlx.h" diff --git a/tests/tests.cpp b/tests/tests.cpp index 0ad8b601..edaf49c2 100644 --- a/tests/tests.cpp +++ b/tests/tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #define DOCTEST_CONFIG_IMPLEMENT #include "doctest/doctest.h" diff --git a/tests/utils_tests.cpp b/tests/utils_tests.cpp index 867a2cf5..35e28adc 100644 --- a/tests/utils_tests.cpp +++ b/tests/utils_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" diff --git a/tests/vmap_tests.cpp b/tests/vmap_tests.cpp index 6ccc25d4..41175b0f 100644 --- a/tests/vmap_tests.cpp +++ b/tests/vmap_tests.cpp @@ -1,3 +1,5 @@ +// Copyright © 2023 Apple Inc. + #include "doctest/doctest.h" #include "mlx/mlx.h"