mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 01:41:17 +08:00
Add set_threadgroup_memory_length to CommandEncoder (#2183)
This commit is contained in:
parent
602f43e3d1
commit
7ff5c41e06
@ -95,6 +95,10 @@ struct CommandEncoder {
|
||||
return enc_->setBytes(&v, sizeof(T), idx);
|
||||
}
|
||||
|
||||
void set_threadgroup_memory_length(size_t length, int idx) {
|
||||
enc_->setThreadgroupMemoryLength(length, idx);
|
||||
}
|
||||
|
||||
ConcurrentContext start_concurrent() {
|
||||
return ConcurrentContext(*this);
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user