From db487e6b1ae08b62dc02d042233f7cbcc78500c3 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Thu, 30 Nov 2023 11:50:36 -0800 Subject: [PATCH] format --- mlx/backend/common/erf.h | 1 - mlx/backend/no_metal/allocator.cpp | 1 - mlx/backend/no_metal/primitives.cpp | 1 - mlx/transforms_impl.h | 1 - python/mlx/utils.py | 1 + tests/utils_tests.cpp | 1 - 6 files changed, 1 insertion(+), 5 deletions(-) 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"