Example usage for org.apache.hadoop.yarn.client RMProxy createRetryPolicy

List of usage examples for org.apache.hadoop.yarn.client RMProxy createRetryPolicy

Introduction

In this page you can find the example usage for org.apache.hadoop.yarn.client RMProxy createRetryPolicy.

Prototype

@Private
@VisibleForTesting
public static RetryPolicy createRetryPolicy(Configuration conf, boolean isHAEnabled) 

Source Link

Document

Fetch retry policy from Configuration

Usage

From source file:com.github.hdl.tensorflow.yarn.app.TFApplicationRpcClient.java

License:Apache License

public TFApplicationRpc getRpc() {
    InetSocketAddress address = new InetSocketAddress(serverAddress, serverPort);
    Configuration conf = new Configuration();
    RetryPolicy retryPolicy = RMProxy.createRetryPolicy(conf, false);
    try {//from   ww  w.  j  a v a  2  s . c o m
        TensorFlowCluster proxy = RMProxy.createRMProxy(conf, TensorFlowCluster.class, address);
        this.tensorflow = (TensorFlowCluster) RetryProxy.create(TensorFlowCluster.class, proxy, retryPolicy);
        return this;
    } catch (IOException e) {
        return null;
    }
}

From source file:org.hdl.caffe.yarn.app.CaffeApplicationRpcClient.java

License:Apache License

public CaffeApplicationRpc getRpc() {
    InetSocketAddress address = new InetSocketAddress(serverAddress, serverPort);
    Configuration conf = new Configuration();
    RetryPolicy retryPolicy = RMProxy.createRetryPolicy(conf, false);
    try {//from   w  w  w  .jav a  2 s.  c o  m
        CaffeCluster proxy = RMProxy.createRMProxy(conf, CaffeCluster.class, address);
        this.caffeCluster = (CaffeCluster) RetryProxy.create(CaffeCluster.class, proxy, retryPolicy);
        return this;
    } catch (IOException e) {
        return null;
    }
}

From source file:org.hdl.tensorflow.yarn.rpc.impl.TFApplicationRpcClient.java

License:Apache License

private TFApplicationRpcClient(String serverAddress, int serverPort) throws IOException {
    InetSocketAddress address = new InetSocketAddress(serverAddress, serverPort);
    Configuration conf = new Configuration();
    RetryPolicy retryPolicy = RMProxy.createRetryPolicy(conf, false);
    TensorFlowCluster proxy = RMProxy.createRMProxy(conf, TensorFlowCluster.class, address);
    this.tensorflow = (TensorFlowCluster) RetryProxy.create(TensorFlowCluster.class, proxy, retryPolicy);
}