zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-12 19:58:11 +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
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.
namespace mlx::core {
/* Approximation to the inverse error function.

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

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

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

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

1
mlx/transforms_impl.h
View File

@@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc.
namespace mlx::core::detail {
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.
def tree_map(fn, tree, *rest):
"""Applies ``fn`` to the leaves of the python tree ``tree`` and
returns a new collection with the results.

1
tests/utils_tests.cpp
View File

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