diff --git hadoop-yarn-project/hadoop-yarn/conf/yarn-env.sh hadoop-yarn-project/hadoop-yarn/conf/yarn-env.sh index 2a40f10..ed9db51 100644 --- hadoop-yarn-project/hadoop-yarn/conf/yarn-env.sh +++ hadoop-yarn-project/hadoop-yarn/conf/yarn-env.sh @@ -34,12 +34,34 @@ fi JAVA=$JAVA_HOME/bin/java JAVA_HEAP_MAX=-Xmx1000m +# For setting YARN specific HEAP sizes please use this +# Parameter and set appropriately +#YARN_HEAPSIZE=1000 + # check envvars which might override default args if [ "$YARN_HEAPSIZE" != "" ]; then #echo "run with heapsize $YARN_HEAPSIZE" JAVA_HEAP_MAX="-Xmx""$YARN_HEAPSIZE""m" #echo $JAVA_HEAP_MAX +else + y=${JAVA_HEAP_MAX#*Xmx} + z=${y%m*} + YARN_HEAPSIZE=$z +fi + +# check envvars which might override default args +if [ "$YARN_RESOURCEMANAGER_HEAPSIZE" = "" ]; then + YARN_RESOURCEMANAGER_HEAPSIZE=$YARN_HEAPSIZE +fi +export YARN_RESOURCEMANAGER_HEAPSIZE="$YARN_RESOURCEMANAGER_HEAPSIZE" +export YARN_RESOURCEMANAGER_OPTS="$YARN_RESOURCEMANAGER_OPTS" + +# check envvars which might override default args +if [ "$YARN_NODEMANAGER_HEAPSIZE" = "" ]; then + YARN_NODEMANAGER_HEAPSIZE=$YARN_HEAPSIZE fi +export YARN_NODEMANAGER_HEAPSIZE="$YARN_NODEMANAGER_HEAPSIZE" +export YARN_NODEMANAGER_OPTS="$YARN_NODEMANAGER_OPTS" # so that filenames w/ spaces are handled correctly in loops below IFS=