mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-19 19:38:16 +08:00
fix multi output leak (#1548)
This commit is contained in:
@@ -271,6 +271,9 @@ array::ArrayDesc::~ArrayDesc() {
|
||||
for (array& a : ad.inputs) {
|
||||
if (a.array_desc_) {
|
||||
input_map.insert({a.id(), a});
|
||||
for (auto& s : a.siblings()) {
|
||||
input_map.insert({s.id(), s});
|
||||
}
|
||||
}
|
||||
}
|
||||
ad.inputs.clear();
|
||||
|
Reference in New Issue
Block a user