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

fix bw for elementwise ops (#2151)

Browse Source 
* fix bw for elementwise ops

* add compile

* fix

* fix

* fix

* fix
This commit is contained in:
Awni Hannun
2025-05-05 06:15:04 -07:00
committed by GitHub
parent 9c5e7da507
commit 825124af8f
12 changed files with 232 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/backend/metal/utils.h
View File

@@ -84,4 +84,12 @@ void concatenate(std::string& acc, T first, Args... args) {
concatenate(acc, args...);
}
inline int get_work_per_thread(Dtype dtype) {
return std::max(1, 8 / dtype.size());
}
inline size_t ceildiv(size_t n, size_t m) {
return (n + m - 1) / m;
}
} // namespace mlx::core