Removes the retain_graph flag (#385)

* Adds global tracing flag
* Removes retain_graph in favor of is_tracer
This commit is contained in:
Angelos Katharopoulos
2024-01-07 15:16:51 -08:00
committed by GitHub
parent 449b43762e
commit a611b0bc82
22 changed files with 209 additions and 207 deletions

View File

@@ -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};
};