zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-26 02:33:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

format

Browse Source
This commit is contained in:
Awni Hannun 2023-11-30 11:50:36 -08:00
parent 266c4e3df6
commit db487e6b1a
6 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mlx/backend/common/erf.h
View File

@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc. // Copyright © 2023 Apple Inc.
namespace mlx::core { namespace mlx::core {
/* Approximation to the inverse error function. /* Approximation to the inverse error function.

1
mlx/backend/no_metal/allocator.cpp
View File

@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc. // Copyright © 2023 Apple Inc.
#include "mlx/allocator.h" #include "mlx/allocator.h"
namespace mlx::core::allocator { namespace mlx::core::allocator {

1
mlx/backend/no_metal/primitives.cpp
View File

@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc. // Copyright © 2023 Apple Inc.
#include "mlx/primitives.h" #include "mlx/primitives.h"
#define NO_GPU(func) \ #define NO_GPU(func) \

1
mlx/transforms_impl.h
View File

@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc. // Copyright © 2023 Apple Inc.
namespace mlx::core::detail { namespace mlx::core::detail {
std::pair<std::vector<array>, std::vector<array>> vmap_trace( std::pair<std::vector<array>, std::vector<array>> vmap_trace(

1
python/mlx/utils.py
View File

@ -1,5 +1,6 @@
# Copyright © 2023 Apple Inc. # Copyright © 2023 Apple Inc.
def tree_map(fn, tree, *rest): def tree_map(fn, tree, *rest):
"""Applies ``fn`` to the leaves of the python tree ``tree`` and """Applies ``fn`` to the leaves of the python tree ``tree`` and
returns a new collection with the results. returns a new collection with the results.

1
tests/utils_tests.cpp
View File

@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc. // Copyright © 2023 Apple Inc.
#include "doctest/doctest.h" #include "doctest/doctest.h"
#include "mlx/mlx.h" #include "mlx/mlx.h"