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

Remove depth traversal (#813)

Browse Source 
* no depth traversal

* counter outside loop
This commit is contained in:
Awni Hannun
2024-03-09 20:21:32 -08:00
committed by GitHub
parent 28301807c2
commit a4d290adb9
3 changed files with 17 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/array.h
View File

@@ -273,11 +273,6 @@ class array {
return outputs;
};
/** The depth of the array in the graph. Evaluated arrays have depth 0. */
uint32_t graph_depth() const {
return array_desc_->depth;
}
/** Detach the array from the graph. */
void detach();
@@ -388,9 +383,6 @@ class array {
// The arrays position in the output list
uint32_t position{0};
// The depth of the array in the graph.
uint32_t depth{0};
explicit ArrayDesc(const std::vector<int>& shape, Dtype dtype);
explicit ArrayDesc(