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

Remove duplicate private section of thread_pool()

Browse Source
This commit is contained in:
Hanyu Deng
2025-04-23 11:00:57 +08:00
committed by Awni Hannun
parent 27ceff33d3
commit f034b51e38
3 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/io/load.h
View File

@@ -99,11 +99,9 @@ class ParallelFileReader : public Reader {
return "file " + label_;
}
private:
static ThreadPool& thread_pool();
private:
static constexpr size_t batch_size_ = 1 << 25;
static ThreadPool& thread_pool();
int fd_;
std::string label_;
};

4
mlx/transforms.cpp
View File

@@ -42,8 +42,8 @@ class Synchronizer : public Primitive {
// are currently under a function transformation and the retain_graph()
// function which returns true if we are forced to retain the graph during
// evaluation.
std::vector<char>& detail::InTracing::trace_stack() {
static std::vector<char> trace_stack_;
std::vector<std::pair<char, char>>& detail::InTracing::trace_stack() {
static std::vector<std::pair<char, char>> trace_stack_;
return trace_stack_;
}
int detail::InTracing::grad_counter{0};

4
mlx/transforms_impl.h
View File

@@ -26,7 +26,7 @@ struct InTracing {
}
~InTracing() {
grad_counter -= trace_stack().back().second;
trace_stack.pop_back();
trace_stack().pop_back();
}
static bool in_tracing() {
@@ -43,7 +43,7 @@ struct InTracing {
private:
static int grad_counter;
static std::vector<char>& trace_stack();
static std::vector<std::pair<char, char>>& trace_stack();
};
struct RetainGraph {