zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-19 10:48:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix bug with multiple attributes (#1348)

Browse Source 
Co-authored-by: Alex Barron <abarron22@apple.com>
This commit is contained in:
Alex Barron
2024-08-23 10:06:15 -07:00
committed by GitHub
parent 98b6ce3460
commit da8deb2b62
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/fast.cpp
View File

@@ -1052,6 +1052,7 @@ void write_signature(
index++;
}
// Add metal attributes e.g. `threadgroup_index_in_grid`
index = 0;
for (const auto& [attr, dtype] : attrs) {
kernel_source << " " << dtype << " " << attr << " [[" << attr << "]]";
if (index < attrs.size() - 1) {
@@ -1059,6 +1060,7 @@ void write_signature(
} else {
kernel_source << ") {" << std::endl;
}
index++;
}
kernel_source << source << std::endl;
kernel_source << "}" << std::endl;