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

throw for certain cases of non captured inputs in compile (#1401)

Browse Source
This commit is contained in:
Awni Hannun
2024-09-09 14:54:31 -07:00
committed by GitHub
parent dc627dcb5e
commit 3ae6aabe9f
5 changed files with 70 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/fast.cpp
View File

@@ -972,7 +972,7 @@ void write_signature(
{"threadgroups_per_grid", "uint3"},
{"threads_per_grid", "uint3"},
{"threads_per_simdgroup", "uint"},
{"thread_per_threadgroup", "uint3"},
{"threads_per_threadgroup", "uint3"},
};
std::vector<std::pair<std::string, std::string>> attrs;
for (const auto& [attr, dtype] : metal_attributes) {