1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
//! This file has been automatically generated by `objc2`'s `header-translator`.
//! DO NOT EDIT
use core::ffi::*;
use core::ptr::NonNull;
use objc2::__framework_prelude::*;
use objc2_foundation::*;
use objc2_metal::*;
use crate::*;
extern_class!(
/// [Apple's documentation](https://developer.apple.com/documentation/metalperformanceshaders/mpsstateresourcelist?language=objc)
#[unsafe(super(NSObject))]
#[derive(Debug, PartialEq, Eq, Hash)]
pub struct MPSStateResourceList;
);
extern_conformance!(
unsafe impl NSObjectProtocol for MPSStateResourceList {}
);
impl MPSStateResourceList {
extern_methods!(
/// Init an empty autoreleased resource list
#[unsafe(method(resourceList))]
#[unsafe(method_family = none)]
pub unsafe fn resourceList() -> Retained<Self>;
/// Init an empty list
#[unsafe(method(init))]
#[unsafe(method_family = init)]
pub unsafe fn init(this: Allocated<Self>) -> Retained<Self>;
/// append a texture to the resource list
#[unsafe(method(appendTexture:))]
#[unsafe(method_family = none)]
pub unsafe fn appendTexture(&self, descriptor: &MTLTextureDescriptor);
/// append a buffer to the resource list
#[unsafe(method(appendBuffer:))]
#[unsafe(method_family = none)]
pub unsafe fn appendBuffer(&self, size: NSUInteger);
);
}
/// Methods declared on superclass `NSObject`.
impl MPSStateResourceList {
extern_methods!(
#[unsafe(method(new))]
#[unsafe(method_family = new)]
pub unsafe fn new() -> Retained<Self>;
);
}
/// [Apple's documentation](https://developer.apple.com/documentation/metalperformanceshaders/mpsstatetextureinfo?language=objc)
#[repr(C)]
#[derive(Clone, Copy, Debug, PartialEq)]
pub struct MPSStateTextureInfo {
pub width: NSUInteger,
pub height: NSUInteger,
pub depth: NSUInteger,
pub arrayLength: NSUInteger,
pub pixelFormat: MTLPixelFormat,
pub textureType: MTLTextureType,
pub usage: MTLTextureUsage,
pub(crate) _reserved: [NSUInteger; 4],
}
unsafe impl Encode for MPSStateTextureInfo {
const ENCODING: Encoding = Encoding::Struct(
"MPSStateTextureInfo",
&[
<NSUInteger>::ENCODING,
<NSUInteger>::ENCODING,
<NSUInteger>::ENCODING,
<NSUInteger>::ENCODING,
<MTLPixelFormat>::ENCODING,
<MTLTextureType>::ENCODING,
<MTLTextureUsage>::ENCODING,
<[NSUInteger; 4]>::ENCODING,
],
);
}
unsafe impl RefEncode for MPSStateTextureInfo {
const ENCODING_REF: Encoding = Encoding::Pointer(&Self::ENCODING);
}
/// [Apple's documentation](https://developer.apple.com/documentation/metalperformanceshaders/mpsstateresourcetype?language=objc)
// NS_ENUM
#[repr(transparent)]
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub struct MPSStateResourceType(pub NSUInteger);
impl MPSStateResourceType {
#[doc(alias = "MPSStateResourceTypeNone")]
pub const None: Self = Self(0);
#[doc(alias = "MPSStateResourceTypeBuffer")]
pub const Buffer: Self = Self(1);
#[doc(alias = "MPSStateResourceTypeTexture")]
pub const Texture: Self = Self(2);
}
unsafe impl Encode for MPSStateResourceType {
const ENCODING: Encoding = NSUInteger::ENCODING;
}
unsafe impl RefEncode for MPSStateResourceType {
const ENCODING_REF: Encoding = Encoding::Pointer(&Self::ENCODING);
}
extern_class!(
/// Dependencies: This depends on Metal Framework
///
/// A semi-opaque data container for large storage in MPS CNN filters
///
/// Some MPS CNN kernels produce additional information beyond a
/// MPSImage. These may be pooling indices where the result came from,
/// convolution weights, or other information not contained in the
/// usual MPSImage result from a MPSCNNKernel. A MPSState object
/// typically contains one or more expensive MTLResources such as
/// textures or buffers to store this information. It provides a
/// base class with interfaces for managing this storage. Child
/// classes may add additional functionality specific to their
/// contents.
///
/// Some MPSState objects are temporary. Temporary state objects,
/// like MPSTemporaryImages and Matrices, are for very short lived storage,
/// perhaps just a few lines of code within the scope of a single
/// MTLCommandBuffer. They are very efficient for storage, as several
/// temporary objects can share the same memory over the course of a
/// MTLCommandBuffer. This can improve both memory usage and time spent
/// in the kernel wiring down memory and such. You may find that some
/// large CNN tasks can not be computed without them, as non-temporary
/// storage would simply take up too much memory.
///
/// In exchange, the lifetime of the underlying storage in temporary
/// MPSState objects needs to be carefully managed. ARC often waits
/// until the end of scope to release objects. Temporary storage often
/// needs to be released sooner than that. Consequently the lifetime of
/// the data in the underlying MTLResources is managed by a readCount
/// property. Each time a MPSCNNKernel reads a temporary MPSState object
/// the readCount is automatically decremented. When it reaches zero, the
/// underlying storage is recycled for use by other MPS temporary objects,
/// and the data is becomes undefined. If you need to consume the data
/// multiple times, you should set the readCount to a larger number to
/// prevent the data from becomming undefined. You may set the readCount
/// to 0 yourself to return the storage to MPS, if for any reason, you
/// realize that the MPSState object will no longer be used.
///
/// The contents of a temporary MPSState object are only valid from
/// creation to the time the readCount reaches 0. The data is only valid
/// for the MTLCommandBuffer on which it was created. Non-temporary
/// MPSState objects are valid on any MTLCommandBuffer on the same
/// device until they are released.
///
/// Finally, temporary MPSState objects are complicated to use with blit encoders.
/// Your application should assume that the temporary MPSState is backed by a MTLHeap,
/// and consequently needs a MTLFence to ensure that compute command encoders and other
/// encoders do not trip over one another with heap based memory. MPS will almost never
/// use a blit encoder for this reason. If you do need one, then you will need to make
/// a new compute encoder to block on whatever previous compute encoder last used the
/// heap block. (MPS will not tell you who previously used the heap block. That encoder
/// is almost certainly long dead anyway.) If concurrent encoders are involved, then a
/// barrier might be needed. Within that compute encoder, you will call -updateFence.
/// End the compute encoder, make a blit encoder wait for the fence, do the blit, update
/// a new fence, then make a new compute encoder, wait for the second fence, then you
/// can continue. Possibly the second do-nothing compute encoder needs to be ended so
/// MPS can be called. Frankly, we don't bother with blit encoders and just write a compute
/// operation for copy / clear as needed, or better yet find a way to eliminate the
/// clear / copy pass so we don't have to pay for it. Your application needs to use
/// temporary MPSStates and MPSTemporaryImages. Memory costs skyrocket, otherwise.
/// It is the blit encoder that is hopefully optional. Note: the most common use of a
/// blit encoder, -synchronizeResource: can not encounter this problem because temporary
/// images and states live in GPU private memory and can not be read by the CPU.
///
/// See also [Apple's documentation](https://developer.apple.com/documentation/metalperformanceshaders/mpsstate?language=objc)
#[unsafe(super(NSObject))]
#[derive(Debug, PartialEq, Eq, Hash)]
pub struct MPSState;
);
extern_conformance!(
unsafe impl NSObjectProtocol for MPSState {}
);
impl MPSState {
extern_methods!(
/// Create a MPSState holding a temporary MTLBuffer
///
/// Parameter `cmdBuf`: The command buffer against which the temporary resource is allocated
///
/// Parameter `bufferSize`: The size of the buffer in bytes
#[unsafe(method(temporaryStateWithCommandBuffer:bufferSize:))]
#[unsafe(method_family = none)]
pub unsafe fn temporaryStateWithCommandBuffer_bufferSize(
cmd_buf: &ProtocolObject<dyn MTLCommandBuffer>,
buffer_size: usize,
) -> Retained<Self>;
/// Create a MPSState holding a temporary MTLTexture
///
/// Parameter `cmdBuf`: The command buffer against which the temporary resource is allocated
///
/// Parameter `descriptor`: A descriptor for the new temporary texture
#[unsafe(method(temporaryStateWithCommandBuffer:textureDescriptor:))]
#[unsafe(method_family = none)]
pub unsafe fn temporaryStateWithCommandBuffer_textureDescriptor(
cmd_buf: &ProtocolObject<dyn MTLCommandBuffer>,
descriptor: &MTLTextureDescriptor,
) -> Retained<Self>;
/// Create a new autoreleased temporary state object without underlying resource
///
/// Parameter `cmdBuf`: The command buffer with which the temporary resource is associated
#[unsafe(method(temporaryStateWithCommandBuffer:))]
#[unsafe(method_family = none)]
pub unsafe fn temporaryStateWithCommandBuffer(
cmd_buf: &ProtocolObject<dyn MTLCommandBuffer>,
) -> Retained<Self>;
#[unsafe(method(initWithDevice:bufferSize:))]
#[unsafe(method_family = init)]
pub unsafe fn initWithDevice_bufferSize(
this: Allocated<Self>,
device: &ProtocolObject<dyn MTLDevice>,
buffer_size: usize,
) -> Retained<Self>;
#[unsafe(method(initWithDevice:textureDescriptor:))]
#[unsafe(method_family = init)]
pub unsafe fn initWithDevice_textureDescriptor(
this: Allocated<Self>,
device: &ProtocolObject<dyn MTLDevice>,
descriptor: &MTLTextureDescriptor,
) -> Retained<Self>;
/// Create a MPSState with a non-temporary MTLResource
///
/// Parameter `resource`: A MTLBuffer or MTLTexture. May be nil.
///
/// # Safety
///
/// - `resource` may need to be synchronized.
/// - `resource` may be unretained, you must ensure it is kept alive while in use.
#[unsafe(method(initWithResource:))]
#[unsafe(method_family = init)]
pub unsafe fn initWithResource(
this: Allocated<Self>,
resource: Option<&ProtocolObject<dyn MTLResource>>,
) -> Retained<Self>;
#[unsafe(method(init))]
#[unsafe(method_family = init)]
pub unsafe fn init(this: Allocated<Self>) -> Option<Retained<Self>>;
/// Initialize a non-temporary state to hold a number of textures and buffers
///
/// The allocation of each resource will be deferred until it is needed.
/// This occurs when -resource or -resourceAtIndex: is called.
///
/// Parameter `resourceList`: The list of resources to create.
#[unsafe(method(initWithDevice:resourceList:))]
#[unsafe(method_family = init)]
pub unsafe fn initWithDevice_resourceList(
this: Allocated<Self>,
device: &ProtocolObject<dyn MTLDevice>,
resource_list: &MPSStateResourceList,
) -> Retained<Self>;
/// Initialize a temporary state to hold a number of textures and buffers
///
/// The textures occur first in sequence
#[unsafe(method(temporaryStateWithCommandBuffer:resourceList:))]
#[unsafe(method_family = none)]
pub unsafe fn temporaryStateWithCommandBuffer_resourceList(
command_buffer: &ProtocolObject<dyn MTLCommandBuffer>,
resource_list: &MPSStateResourceList,
) -> Retained<Self>;
/// Create a state object with a list of MTLResources
///
/// Because MPS prefers deferred allocation of resources
/// your application should use -initWithTextures:bufferSizes:bufferCount:
/// whenever possible. This method is useful for cases when the
/// MTLResources must be initialized by the CPU.
///
/// # Safety
///
/// - `resources` generic may need to be synchronized.
/// - `resources` generic may be unretained, you must ensure it is kept alive while in use.
#[unsafe(method(initWithResources:))]
#[unsafe(method_family = init)]
pub unsafe fn initWithResources(
this: Allocated<Self>,
resources: Option<&NSArray<ProtocolObject<dyn MTLResource>>>,
) -> Retained<Self>;
/// Return the number of MTLResource objects held by the state
#[unsafe(method(resourceCount))]
#[unsafe(method_family = none)]
pub unsafe fn resourceCount(&self) -> NSUInteger;
/// Get the MTLResource at the indicated index
///
/// By convention, except where otherwise documented, the MTLResources
/// held by the MPSState are private to the MPSState object, owned
/// by the MPSState subclass author. If the MPSState subclass
/// author is MPS, then the identity (e.g. texture vs. buffer)
/// and information contained in the resource should be considered
/// implementation dependent. It may change by operating system
/// version or device. If you are the author of the subclass then it
/// is for your own use, and MPS will not look at it, except perhaps
/// so as to pass it to a custom kernel. Otherwise, the method is made
/// available to facilitate debugging and to allow you to write your own
/// state objects. Provide accessors to read this information
/// in a defined format.
///
///
/// Parameter `index`: The index of the MTLResource to retrieve
///
/// Parameter `allocateMemory`: It is very important to avoid allocating memory to hold
/// MTLResources until it is absolutely necessary, especially when working
/// with temporary MPSStates. When allocateMemory is set to NO and the
/// resource has not yet been allocated, nil will be returned instead.
/// If you just need information about the resource such as buffer size
/// or MTLTexture properties, but not the resource itself, please use
/// -bufferSizeAtIndex: or -textureInfoAtIndex: instead, as these will
/// not force the creation of the MTLResource.
#[unsafe(method(resourceAtIndex:allocateMemory:))]
#[unsafe(method_family = none)]
pub unsafe fn resourceAtIndex_allocateMemory(
&self,
index: NSUInteger,
allocate_memory: bool,
) -> Option<Retained<ProtocolObject<dyn MTLResource>>>;
#[unsafe(method(readCount))]
#[unsafe(method_family = none)]
pub unsafe fn readCount(&self) -> NSUInteger;
/// Setter for [`readCount`][Self::readCount].
#[unsafe(method(setReadCount:))]
#[unsafe(method_family = none)]
pub unsafe fn setReadCount(&self, read_count: NSUInteger);
#[unsafe(method(isTemporary))]
#[unsafe(method_family = none)]
pub unsafe fn isTemporary(&self) -> bool;
/// A string to help identify this object.
#[unsafe(method(label))]
#[unsafe(method_family = none)]
pub unsafe fn label(&self) -> Option<Retained<NSString>>;
/// Setter for [`label`][Self::label].
///
/// This is [copied][objc2_foundation::NSCopying::copy] when set.
#[unsafe(method(setLabel:))]
#[unsafe(method_family = none)]
pub unsafe fn setLabel(&self, label: Option<&NSString>);
/// Return the buffer size of the MTLBuffer at index or 0 if it is not a MTLBuffer
///
/// Does not force allocation of the MTLResource
#[unsafe(method(bufferSizeAtIndex:))]
#[unsafe(method_family = none)]
pub unsafe fn bufferSizeAtIndex(&self, index: NSUInteger) -> NSUInteger;
/// Return the texture size {width,height,depth} or {0,0,0} if it is not a MTLTexture
///
/// Does not force allocation of the MTLResource
#[unsafe(method(textureInfoAtIndex:))]
#[unsafe(method_family = none)]
pub unsafe fn textureInfoAtIndex(&self, index: NSUInteger) -> MPSStateTextureInfo;
/// Return YES if the resource at index is a buffer
///
/// Does not force allocation of the MTLResource
#[unsafe(method(resourceTypeAtIndex:))]
#[unsafe(method_family = none)]
pub unsafe fn resourceTypeAtIndex(&self, index: NSUInteger) -> MPSStateResourceType;
/// Flush any copy of MTLResources held by the state from the device's caches, and invalidate any CPU caches if needed.
///
/// This will call [id
/// <MTLBlitEncoder
/// > synchronizeResource: ] on the state's MTLResources.
/// This is necessary for all MTLStorageModeManaged resources. For other resources, including temporary
/// resources (these are all MTLStorageModePrivate), nothing is done.
///
/// Parameter `commandBuffer`: The commandbuffer on which to synchronize
#[unsafe(method(synchronizeOnCommandBuffer:))]
#[unsafe(method_family = none)]
pub unsafe fn synchronizeOnCommandBuffer(
&self,
command_buffer: &ProtocolObject<dyn MTLCommandBuffer>,
);
/// Get the number of bytes used to allocate underyling MTLResources
///
/// This is the size of the backing store of underlying MTLResources.
/// It does not include all storage used by the object, for example
/// the storage used to hold the MPSState instantiation and MTLTexture
/// or MTLBuffer is not included. It only measures the size of the
/// allocation used to hold the texels in the texture or bytes in the
/// buffer. This value is subject to change between different devices
/// and operating systems.
///
/// Except when -initWithResource: is used, most MPSStates are allocated
/// without a backing store. The backing store is allocated lazily when
/// it is needed, typically when the .texture property is called.
/// Consequently, in most cases, it should be inexpensive to make
/// a MPSImage to see how much memory it will need, and release it
/// if it is too large.
///
/// This method may fail in certain circumstances, such as when the
/// MPSImage is created with -initWithTexture:featureChannels:, in
/// which case 0 will be returned.
#[unsafe(method(resourceSize))]
#[unsafe(method_family = none)]
pub unsafe fn resourceSize(&self) -> NSUInteger;
#[cfg(all(feature = "MPSImage", feature = "MPSKernel"))]
/// Determine padding and sizing of result images
///
/// A MPSState has the opportunity to reconfigure the MPSImageDescriptor
/// used to create the filter result state and set the MPSKernel.offset
/// to the correct value. By default, the MPSState does not modify the
/// descriptor.
///
/// There is a order of operations defined for who may update the descriptor:
///
/// 1) Default padding code runs based on the MPSNNPaddingMethod in
/// the MPSCNNKernel.padding. This creates the descriptor and
/// picks a starting value for the MPSCNNKernel.offset.
/// 2) MPSStates are called in order to apply this function and update
/// the offset.
/// 3) The MPSNNPadding custom padding method of the same name is called.
/// 4)
///
/// Some code that may prove helpful:
///
///
/// ```text
/// const int centeringPolicy = 0; // When kernelSize is even: 0 pad bottom right. 1 pad top left. Centers the kernel for even sized kernels.
///
/// typedef enum Style{
/// StyleValidOnly = -1,
/// StyleSame = 0,
/// StyleFull = 1
/// }Style;
///
/// // Typical destination size in one dimension for forward filters (most filters)
/// static int DestSize( int sourceSize, int stride, int filterWindowSize, Style style ){
/// sourceSize += style * (filterWindowSize - 1); // adjust how many pixels we are allowed to read
/// return (sourceSize + stride - 1) / stride; // sourceSize / stride, round up
/// }
///
/// // Typical destination size in one dimension for reverse filters (e.g. convolution transpose)
/// static int DestSizeReverse( int sourceSize, int stride, int filterWindowSize, Style style ){
/// return (sourceSize-1) * stride + // center tap for the last N-1 results. Take stride into account
/// 1 + // center tap for the first result
/// style * (filterWindowSize-1); // add or subtract (or ignore) the filter extent
/// }
///
/// // Find the MPSOffset in one dimension
/// static int Offset( int sourceSize, int stride, int filterWindowSize, Style style ){
/// // The correction needed to adjust from position of left edge to center per MPSOffset definition
/// int correction = filterWindowSize / 2;
///
/// // exit if all we want is to start consuming pixels at the left edge of the image.
/// if( 0 )
/// return correction;
///
/// // Center the area consumed in the source image:
/// // Calculate the size of the destination image
/// int destSize = DestSize( sourceSize, stride, filterWindowSize, style ); // use DestSizeReverse here instead as appropriate
///
/// // calculate extent of pixels we need to read in source to populate the destination
/// int readSize = (destSize-1) * stride + filterWindowSize;
///
/// // calculate number of missing pixels in source
/// int extraSize = readSize - sourceSize;
///
/// // number of missing pixels on left side
/// int leftExtraPixels = (extraSize + centeringPolicy) / 2;
///
/// // account for the fact that the offset is based on the center pixel, not the left edge
/// return correction - leftExtraPixels;
/// }
/// ```
///
///
/// Parameter `sourceImages`: The list of source images to be used
///
/// Parameter `sourceStates`: The list of source states to be used
///
/// Parameter `kernel`: The MPSKernel the padding method will be applied to. Set the kernel.offset
///
/// Parameter `inDescriptor`: MPS will prepare a starting guess based on the padding policy (exclusive of
/// MPSNNPaddingMethodCustom) set for the object. You should adjust the offset
/// and image size accordingly. It is on an autoreleasepool.
///
///
/// Returns: The MPSImageDescriptor to use to make a MPSImage to capture the results from the filter.
/// The MPSImageDescriptor is assumed to be on an autoreleasepool. Your method must also set the
/// kernel.offset property.
#[unsafe(method(destinationImageDescriptorForSourceImages:sourceStates:forKernel:suggestedDescriptor:))]
#[unsafe(method_family = none)]
pub unsafe fn destinationImageDescriptorForSourceImages_sourceStates_forKernel_suggestedDescriptor(
&self,
source_images: &NSArray<MPSImage>,
source_states: Option<&NSArray<MPSState>>,
kernel: &MPSKernel,
in_descriptor: &MPSImageDescriptor,
) -> Retained<MPSImageDescriptor>;
/// Get the private MTLResource underlying the MPSState
///
/// When the state is not directly initialized with a MTLResource,
/// the actuall MTLResource creation is deferred. Especially with
/// temporary resources, it is important to delay this creation
/// as late as possible to avoid increasing the memory footprint.
/// The memory is returned for reuse when the readCount = 0. Calling
/// the -resource method will force the resource to be allocated,
/// so you should not use it lightly, for purposes such as finding
/// the MTLPixelFormat of a texture in the state.
///
/// By convention, except where otherwise documented, the MTLResources
/// held by the MPSState are private to the MPSState object, owned
/// by the MPSState subclass author. If the MPSState subclass
/// author is MPS, then the identity (e.g. texture vs. buffer)
/// and information contained in the resource should be considered
/// implementation dependent. It may change by operating system
/// version or device. If you are the author of the subclass then it
/// is for your own use, and MPS will not look at it, except perhaps
/// so as to pass it to a custom kernel. Otherwise, the method is made
/// available to facilitate debugging and to allow you to write your own
/// state objects.
#[deprecated]
#[unsafe(method(resource))]
#[unsafe(method_family = none)]
pub unsafe fn resource(&self) -> Option<Retained<ProtocolObject<dyn MTLResource>>>;
);
}
/// Methods declared on superclass `NSObject`.
impl MPSState {
extern_methods!(
#[unsafe(method(new))]
#[unsafe(method_family = new)]
pub unsafe fn new() -> Retained<Self>;
);
}
/// [Apple's documentation](https://developer.apple.com/documentation/metalperformanceshaders/mpsstatebatch?language=objc)
pub type MPSStateBatch = NSArray<MPSState>;
impl MPSState {
/// raise or lower the readcount of a batch by a set amount
///
/// In some circumstances, a MPSState may appear in a MPSStateBatch
/// multiple times. This is particularly common when the MPSState serves
/// as an accumulator across the entire batch, such as when accumulating
/// gradients for convolution weight update or batch statistics for
/// batch normalization. A naive function would then end up incrementing
/// the state multiple times, probably leading to an error.
///
/// MPSStateBatchIncrementReadCount() will efficiently increment the
/// readCounts of each object in the batch only once, avoiding this problem.
///
///
/// Parameter `batch`: The MPSStateBatch to increment
///
/// Parameter `amount`: The value to add to the read count for each unique state in the batch
///
/// Returns: The number of different objects in the batch
#[doc(alias = "MPSStateBatchIncrementReadCount")]
#[inline]
pub unsafe fn batch_increment_read_count(
batch: Option<&MPSStateBatch>,
amount: NSInteger,
) -> NSUInteger {
extern "C-unwind" {
fn MPSStateBatchIncrementReadCount(
batch: Option<&MPSStateBatch>,
amount: NSInteger,
) -> NSUInteger;
}
unsafe { MPSStateBatchIncrementReadCount(batch, amount) }
}
/// Call [MTLBlitEncoder synchronizeResource:] on unique resources
#[doc(alias = "MPSStateBatchSynchronize")]
#[inline]
pub unsafe fn batch_synchronize(
batch: &MPSStateBatch,
cmd_buf: &ProtocolObject<dyn MTLCommandBuffer>,
) {
extern "C-unwind" {
fn MPSStateBatchSynchronize(
batch: &MPSStateBatch,
cmd_buf: &ProtocolObject<dyn MTLCommandBuffer>,
);
}
unsafe { MPSStateBatchSynchronize(batch, cmd_buf) }
}
/// Call [MTLBlitEncoder resourceSize] on unique resources
#[doc(alias = "MPSStateBatchResourceSize")]
#[inline]
pub unsafe fn batch_resource_size(batch: Option<&MPSStateBatch>) -> NSUInteger {
extern "C-unwind" {
fn MPSStateBatchResourceSize(batch: Option<&MPSStateBatch>) -> NSUInteger;
}
unsafe { MPSStateBatchResourceSize(batch) }
}
}
extern "C-unwind" {
#[deprecated = "renamed to `MPSState::batch_increment_read_count`"]
pub fn MPSStateBatchIncrementReadCount(
batch: Option<&MPSStateBatch>,
amount: NSInteger,
) -> NSUInteger;
}
extern "C-unwind" {
#[deprecated = "renamed to `MPSState::batch_synchronize`"]
pub fn MPSStateBatchSynchronize(
batch: &MPSStateBatch,
cmd_buf: &ProtocolObject<dyn MTLCommandBuffer>,
);
}
extern "C-unwind" {
#[deprecated = "renamed to `MPSState::batch_resource_size`"]
pub fn MPSStateBatchResourceSize(batch: Option<&MPSStateBatch>) -> NSUInteger;
}