mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-20 01:11:44 +08:00
Removes the retain_graph
flag (#385)
* Adds global tracing flag * Removes retain_graph in favor of is_tracer
This commit is contained in:

committed by
GitHub

parent
449b43762e
commit
a611b0bc82
@@ -183,7 +183,7 @@ TEST_CASE("test vmap with eval") {
|
||||
auto fun2 = [](std::vector<array> inputs) {
|
||||
auto x = inputs[0] + 1;
|
||||
auto y = inputs[1] + 2;
|
||||
eval({x}, true);
|
||||
eval(x);
|
||||
auto out = add(x, y);
|
||||
return std::vector<array>{out};
|
||||
};
|
||||
|
Reference in New Issue
Block a user