MCPcopy Create free account
hub / github.com/ml-explore/mlx / outer_fun

Function outer_fun

tests/compile_tests.cpp:88–92  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

86}
87
88std::vector<array> outer_fun(const std::vector<array>& inputs) {
89 auto x = inputs[0] + inputs[1];
90 auto y = compile(inner_fun)({x})[0];
91 return {x + y};
92}
93
94TEST_CASE("test nested compile") {
95 auto cfun = compile(outer_fun);

Callers

nothing calls this directly

Calls 1

compileFunction · 0.50

Tested by

no test coverage detected