Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

The definition of ExternalResourceDriver and ExternalResourceInfo is:


Code Block
languagejava
titleExternalResourceDriver
public abstract class ExternalResourceDriver {

...



    /**

...


    * Retrieve the information of the external resources according to the resourceProfile.

...


    */

...


    List<ExternalResourceInfo> retrieveResourceInfo(ResourceProfile resourceProfile);

...



    /**

...


    * When running in Kubernetes, we need to decorate the TM pod to request the external resource.

...


    */

...


    AbstractKubernetesStepDecorator getExternalResourceDecorator();

...



    /**

...


    * When running in Yarn, we need to add the external resource request to the Resource of 

...


    */

...


    void addExternalResourceToRequest(AMRMClient.ContainerRequest containerRequest);

...


}

...



public abstract class ExternalResourceInfo {

...



    // Return the name of that external resource.

...


    String getName();

...



}


We introduce the GPUDriver for the GPU resources.

...