zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats 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);
} }