I think this is possible in Metal by calling `setThreadgroupMemoryLength` to configure `threadgroup` arguments.