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.

A sample 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.