zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix shapeless compile on ubuntu24 (#1775)

Browse Source
This commit is contained in:
Awni Hannun 2025-01-18 06:04:36 -08:00 committed by GitHub
parent a4667da1eb
commit a8666a757a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/transforms.cpp
View File

@ -37,7 +37,7 @@ class Synchronizer : public Primitive {
// are currently under a function transformation and the retain_graph() // are currently under a function transformation and the retain_graph()
// function which returns true if we are forced to retain the graph during // function which returns true if we are forced to retain the graph during
// evaluation. // evaluation.
std::vector<bool> detail::InTracing::trace_stack{}; std::vector<char> detail::InTracing::trace_stack{};
int detail::RetainGraph::tracing_counter{0}; int detail::RetainGraph::tracing_counter{0};
array eval_impl(std::vector<array> outputs, bool async) { array eval_impl(std::vector<array> outputs, bool async) {

2
mlx/transforms_impl.h
View File

@ -36,7 +36,7 @@ struct InTracing {
} }
private: private:
static std::vector<bool> trace_stack; static std::vector<char> trace_stack;
}; };
struct RetainGraph { struct RetainGraph {