diff --git a/mlx/backend/common/erf.h b/mlx/backend/common/erf.h index 25a53d375..a175a0c43 100644 --- a/mlx/backend/common/erf.h +++ b/mlx/backend/common/erf.h @@ -1,6 +1,5 @@ // Copyright © 2023 Apple Inc. - namespace mlx::core { /* Approximation to the inverse error function. diff --git a/mlx/backend/no_metal/allocator.cpp b/mlx/backend/no_metal/allocator.cpp index dcb67d93d..e0d777f4a 100644 --- a/mlx/backend/no_metal/allocator.cpp +++ b/mlx/backend/no_metal/allocator.cpp @@ -1,6 +1,5 @@ // Copyright © 2023 Apple Inc. - #include "mlx/allocator.h" namespace mlx::core::allocator { diff --git a/mlx/backend/no_metal/primitives.cpp b/mlx/backend/no_metal/primitives.cpp index 3c327c677..0d020c26a 100644 --- a/mlx/backend/no_metal/primitives.cpp +++ b/mlx/backend/no_metal/primitives.cpp @@ -1,6 +1,5 @@ // Copyright © 2023 Apple Inc. - #include "mlx/primitives.h" #define NO_GPU(func) \ diff --git a/mlx/transforms_impl.h b/mlx/transforms_impl.h index 0fa8f06a2..201e8009b 100644 --- a/mlx/transforms_impl.h +++ b/mlx/transforms_impl.h @@ -1,6 +1,5 @@ // Copyright © 2023 Apple Inc. - namespace mlx::core::detail { std::pair, std::vector> vmap_trace( diff --git a/python/mlx/utils.py b/python/mlx/utils.py index d70b8ce57..1fb6bd498 100644 --- a/python/mlx/utils.py +++ b/python/mlx/utils.py @@ -1,5 +1,6 @@ # 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/tests/utils_tests.cpp b/tests/utils_tests.cpp index 35e28adc3..4c4cf31ac 100644 --- a/tests/utils_tests.cpp +++ b/tests/utils_tests.cpp @@ -1,6 +1,5 @@ // Copyright © 2023 Apple Inc. - #include "doctest/doctest.h" #include "mlx/mlx.h"