cavis/libnd4j/include/memory/impl/MemoryUtils.cpp

/* ******************************************************************************
 *
 *
 * This program and the accompanying materials are made available under the
 * terms of the Apache License, Version 2.0 which is available at
 * https://www.apache.org/licenses/LICENSE-2.0.
 *
 *  See the NOTICE file distributed with this work for additional
 *  information regarding copyright ownership.
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 ******************************************************************************/

//
// Created by raver119 on 11.10.2017.
//

#include "memory/MemoryUtils.h"
#include <helpers/logger.h>

#if defined(__APPLE__)
#include<mach/mach.h>
#include <sys/resource.h>
#elif defined(_WIN32) || defined(__WIN32__) || defined(WIN32)

#else
// linux
#include <sys/resource.h>
#include <fcntl.h>
#include <unistd.h>
#include <cstring>
#endif


bool sd::memory::MemoryUtils::retrieveMemoryStatistics(sd::memory::MemoryReport &report) {
#if defined(__APPLE__)
    nd4j_debug("APPLE route\n", "");
/*
    struct task_basic_info t_info;
    mach_msg_type_number_t t_info_count = TASK_BASIC_INFO_COUNT;

    if (KERN_SUCCESS != task_info(mach_task_self(), TASK_BASIC_INFO, (task_info_t)&t_info, &t_info_count))
        return false;

    report.setVM(t_info.resident_size);
    report.setRSS(t_info.resident_size);


    nd4j_debug("RSS: %lld; VM: %lld;\n", report.getRSS(), report.getVM());
*/
    struct rusage _usage;

    auto res = getrusage(RUSAGE_SELF, &_usage);

    report.setRSS(_usage.ru_maxrss);

    nd4j_debug("Usage: %lld; %lld; %lld; %lld;\n", _usage.ru_ixrss, _usage.ru_idrss, _usage.ru_isrss, _usage.ru_maxrss);

    return true;
#elif defined(_WIN32) || defined(__WIN32__) || defined(WIN32)
    nd4j_debug("WIN32 route\n", "");


#else
    nd4j_debug("LINUX route\n", "");
    int fd = open("/proc/self/statm", O_RDONLY, 0);
    if (fd >= 0) {
        char line[256];
        char* s;
        int n;
        lseek(fd, 0, SEEK_SET);
        if ((n = read(fd, line, sizeof(line))) > 0 && (s = (char*)memchr(line, ' ', n)) != NULL) {
            report.setRSS((Nd4jLong)(atoll(s + 1) * getpagesize()));
        }
        close(fd);
    }

    /*
    struct rusage _usage;

    auto res = getrusage(RUSAGE_SELF, &_usage);

    report.setRSS(_usage.ru_maxrss);

    //nd4j_printf("Usage: %lld; %lld; %lld; %lld;\n", _usage.ru_ixrss, _usage.ru_idrss, _usage.ru_isrss, _usage.ru_maxrss);
     */


    return true;
#endif

    return false;
}