This commit is contained in:
Awni Hannun
2023-11-30 11:50:50 -08:00
parent 266c4e3df6
commit db487e6b1a
6 changed files with 1 additions and 5 deletions
-1
View File
@@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc.
namespace mlx::core {
/* Approximation to the inverse error function.
-1
View File
@@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc.
#include "mlx/allocator.h"
namespace mlx::core::allocator {
-1
View File
@@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc.
#include "mlx/primitives.h"
#define NO_GPU(func) \
-1
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
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
View File
@@ -1,6 +1,5 @@
// Copyright © 2023 Apple Inc.
#include "doctest/doctest.h"
#include "mlx/mlx.h"