Implement the DevicePluginScheduler interface
You can implement the
DevicePluginScheduler interface if you want to
use the plug-in with a more efficient scheduler. The
allocateDevices method is
invoked by YARN each time when checking the plug-in's recommended devices for one container.
This interface is optional because YARN provides a very basic scheduler.
NvidiaGPUPluginForRuntimeV2 plug-in for a customized scheduler is
available in the framework for your reference. This sample scheduler is for Nvidia GPU
topology and can get considerable performance boost for the container.