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

Add set_threadgroup_memory_length to CommandEncoder (#2183)

Browse Source
This commit is contained in:
Jack Wind
2025-05-16 03:28:03 -04:00
committed by GitHub
parent 602f43e3d1
commit 7ff5c41e06
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -95,6 +95,10 @@ struct CommandEncoder {
return enc_->setBytes(&v, sizeof(T), idx); return enc_->setBytes(&v, sizeof(T), idx);
} }
void set_threadgroup_memory_length(size_t length, int idx) {
enc_->setThreadgroupMemoryLength(length, idx);
}
ConcurrentContext start_concurrent() { ConcurrentContext start_concurrent() {
return ConcurrentContext(*this); return ConcurrentContext(*this);
} }