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

dtype is copy assignable (#1436)

Browse Source
This commit is contained in:
Awni Hannun
2024-09-25 12:07:13 -07:00
committed by GitHub
parent 195b429d99
commit afc9c0ec1b
7 changed files with 29 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mlx/utils.cpp
View File

@@ -58,11 +58,11 @@ inline void PrintFormatter::print(std::ostream& os, complex64_t val) {
PrintFormatter global_formatter;
Dtype result_type(const std::vector<array>& arrays) {
std::vector<Dtype> dtypes(1, bool_);
Dtype t = bool_;
for (auto& arr : arrays) {
dtypes.push_back(promote_types(dtypes.back(), arr.dtype()));
t = promote_types(t, arr.dtype());
}
return dtypes.back();
return t;
}
std::vector<int> broadcast_shapes(