Cette page a été traduite par l'API Cloud Translation.

GPUOptions.ExperimentalOrBuilder

interface statique publique GPUOptions.ExperimentalOrBuilder

Sous-classes indirectes connues

GPUOptions.Experimental , GPUOptions.Experimental.Builder

Méthodes publiques

chaîne abstraite	getCollectiveRingOrder () If non-empty, defines a good GPU ring order on a single worker based on device interconnect.
résumé com.google.protobuf.ByteString	getCollectiveRingOrderBytes () If non-empty, defines a good GPU ring order on a single worker based on device interconnect.
abstrait entier	getKernelTrackerMaxBytes () If kernel_tracker_max_bytes = n > 0, then a tracking event is inserted after every series of kernels allocating a sum of memory >= n.
abstrait entier	getKernelTrackerMaxInterval () Parameters for GPUKernelTracker.
abstrait entier	getKernelTrackerMaxPending () If kernel_tracker_max_pending > 0 then no more than this many tracking events can be outstanding at a time.
abstrait entier	getNumDevToDevCopyStreams () If > 1, the number of device-to-device copy streams to create for each GPUDevice.
booléen abstrait	getTimestampedAllocator () If true then extra work is done by GPUDevice and GPUBFCAllocator to keep track of when GPU memory is freed and when kernels actually complete so that we can know when a nominally free memory chunk is really not subject to pending use.
booléen abstrait	getUseUnifiedMemory () If true, uses CUDA unified memory for memory allocations.
résumé GPUOptions.Experimental.VirtualDevices	getVirtualDevices (index int) The multi virtual device settings.
abstrait entier	getVirtualDevicesCount () The multi virtual device settings.
Liste abstraite < GPUOptions.Experimental.VirtualDevices >	getVirtualDevicesList () The multi virtual device settings.
abstrait GPUOptions.Experimental.VirtualDevicesOrBuilder	getVirtualDevicesOrBuilder (index int) The multi virtual device settings.
Liste abstraite <? étend GPUOptions.Experimental.VirtualDevicesOrBuilder >	getVirtualDevicesOrBuilderList () The multi virtual device settings.

Méthodes publiques

chaîne abstraite publique getCollectiveRingOrder ()

 If non-empty, defines a good GPU ring order on a single worker based on
 device interconnect.  This assumes that all workers have the same GPU
 topology.  Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4".
 This ring order is used by the RingReducer implementation of
 CollectiveReduce, and serves as an override to automatic ring order
 generation in OrderTaskDeviceMap() during CollectiveParam resolution.

string collective_ring_order = 4;

résumé public com.google.protobuf.ByteString getCollectiveRingOrderBytes ()

 If non-empty, defines a good GPU ring order on a single worker based on
 device interconnect.  This assumes that all workers have the same GPU
 topology.  Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4".
 This ring order is used by the RingReducer implementation of
 CollectiveReduce, and serves as an override to automatic ring order
 generation in OrderTaskDeviceMap() during CollectiveParam resolution.

string collective_ring_order = 4;

public abstrait int getKernelTrackerMaxBytes ()

 If kernel_tracker_max_bytes = n > 0, then a tracking event is
 inserted after every series of kernels allocating a sum of
 memory >= n.  If one kernel allocates b * n bytes, then one
 event will be inserted after it, but it will count as b against
 the pending limit.

int32 kernel_tracker_max_bytes = 8;

public abstrait int getKernelTrackerMaxInterval ()

 Parameters for GPUKernelTracker.  By default no kernel tracking is done.
 Note that timestamped_allocator is only effective if some tracking is
 specified.
 If kernel_tracker_max_interval = n > 0, then a tracking event
 is inserted after every n kernels without an event.

int32 kernel_tracker_max_interval = 7;

public abstrait int getKernelTrackerMaxPending ()

 If kernel_tracker_max_pending > 0 then no more than this many
 tracking events can be outstanding at a time.  An attempt to
 launch an additional kernel will stall until an event
 completes.

int32 kernel_tracker_max_pending = 9;

public abstract int getNumDevToDevCopyStreams ()

 If > 1, the number of device-to-device copy streams to create
 for each GPUDevice.  Default value is 0, which is automatically
 converted to 1.

int32 num_dev_to_dev_copy_streams = 3;

public abstrait booléen getTimestampedAllocator ()

 If true then extra work is done by GPUDevice and GPUBFCAllocator to
 keep track of when GPU memory is freed and when kernels actually
 complete so that we can know when a nominally free memory chunk
 is really not subject to pending use.

bool timestamped_allocator = 5;

public abstrait booléen getUseUnifiedMemory ()

 If true, uses CUDA unified memory for memory allocations. If
 per_process_gpu_memory_fraction option is greater than 1.0, then unified
 memory is used regardless of the value for this field. See comments for
 per_process_gpu_memory_fraction field for more details and requirements
 of the unified memory. This option is useful to oversubscribe memory if
 multiple processes are sharing a single GPU while individually using less
 than 1.0 per process memory fraction.

bool use_unified_memory = 2;