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 compile (#1501)

Browse Source 
* fix compile

* fix space
This commit is contained in:
Awni Hannun
2024-10-18 11:06:40 -07:00
committed by GitHub
parent 50d8bed468
commit 92d7cb71f8
5 changed files with 36 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -30,8 +30,8 @@ std::string type_to_name(const array& a);
// Compute the thread block dimensions which fit the given
// input dimensions.
// - The thread block dimensions will be powers of two
// - The thread block size will be less than 1024
MTL::Size get_block_dims(int dim0, int dim1, int dim2);
// - The thread block size will be less than 2^pow2
MTL::Size get_block_dims(int dim0, int dim1, int dim2, int pow2 = 10);
// Computes a 2D grid where each element is < UINT_MAX
// Assumes: