Skip to content

Commit 492ca0b

Browse files
committed
update MemAllocator
1 parent fd35ab1 commit 492ca0b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

rfcs/20200624-pluggable-device-for-tensorflow.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,8 @@ Two sets of classes need to be defined in TensorFlow proper.
226226
* class `PluggableDevice`: a class represents a set of new third-party devices, its device_type attribute describes what kind of device this is. it can be "GPU" or other device type string. it also has an attribute: subdevice_type, subdevice_type is for low-level specialization of GPU device. It will be part of kernel dispatch key to avoid conflict issue with exiting GPU(CUDA) kernels. The subdevice_type is also used to check whether there is some CUDA specific logic code in grappler and common runtime when the device type is "GPU".
227227
* class `PluggableDeviceFactory`: a device factory to create the PluggableDevice
228228
* class `PluggableDeviceBFCAllocator`: a PluggableDevice memory allocator that implements a ‘best fit with coalescing’ algorithm. It extends the BFC algorithm, counter part of GPUBFCAllocator.
229-
* class `PluggableDeviceAllocator`: an allocator that wraps a PluggableDevice allocator.
229+
* class `PluggableDeviceMemAllocator`: Suballocator for GPU memory.
230+
* class `PluggableDeviceProcessState`: manages per-process state when PluggableDevices are present
230231
* class `PluggableDeviceHostAllocator`: allocator for pinned CPU RAM that is made known to PluggableDevice for the purpose of efficient DMA with PluggableDevice.
231232
* class `PluggableDeviceEventMgr`: an object to keep track of pending Events in the StreamExecutor streams.
232233
* class `PluggableDeviceContext`: a wrapper of pluggable device specific context that can be passed to OpKernels.

0 commit comments

Comments
 (0)