001/* 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, 013 * software distributed under the License is distributed on an 014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 015 * KIND, either express or implied. See the License for the 016 * specific language governing permissions and limitations 017 * under the License. 018 */ 019package org.apache.reef.examples.group.bgd; 020 021import org.apache.reef.client.LauncherStatus; 022import org.apache.reef.examples.group.utils.timer.Timer; 023import org.apache.reef.runtime.yarn.client.YarnClientConfiguration; 024import org.apache.reef.tang.Configuration; 025 026import java.util.logging.Level; 027import java.util.logging.Logger; 028 029/** 030 * Runs BGD on the YARN runtime. 031 */ 032public final class BGDYarn { 033 034 private static final Logger LOG = Logger.getLogger(BGDYarn.class.getName()); 035 036 private static final int TIMEOUT = 4 * Timer.HOURS; 037 038 public static void main(final String[] args) throws Exception { 039 040 final BGDClient bgdClient = BGDClient.fromCommandLine(args); 041 042 final Configuration runtimeConfiguration = YarnClientConfiguration.CONF 043 .set(YarnClientConfiguration.JVM_HEAP_SLACK, "0.1") 044 .build(); 045 046 final String jobName = System.getProperty("user.name") + "-" + "BR-ResourceAwareBGD-YARN"; 047 048 final LauncherStatus status = bgdClient.run(runtimeConfiguration, jobName, TIMEOUT); 049 050 LOG.log(Level.INFO, "OUT: Status = {0}", status); 051 } 052 053 /** 054 * Empty private constructor to prohibit instantiation of utility class. 055 */ 056 private BGDYarn() { 057 } 058}