Sha256: 073440c6423cbbe46b65e7f0106d66e63949c1f36382caf0959fa5d1e32c4622
Contents?: true
Size: 865 Bytes
Versions: 7
Compression:
Stored size: 865 Bytes
Contents
#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif
#include <sched.h>
#include "ruby.h"
#if defined(__GNUC__) && !defined(__clang__) && !defined(__INTEL_COMPILER)
#pragma GCC optimize ("O3")
#pragma GCC diagnostic warning "-Wall"
#elif defined(__clang__)
#pragma clang optimize on
#pragma clang diagnostic warning "-Wall"
#elif defined(__INTEL_COMPILER)
#pragma intel optimization_level 3
#endif
static VALUE count_cpu_for_pid(volatile VALUE obj, volatile VALUE pid) {
cpu_set_t set ;
CPU_ZERO(&set) ;
char status = sched_getaffinity(FIX2INT(pid), sizeof(set), &set) ;
if (status < 0) return Qnil ;
return INT2FIX(CPU_COUNT(&set)) ;
}
void Init_nproc() {
VALUE _linux_stat = rb_define_module("LinuxStat") ;
VALUE _nproc = rb_define_module_under(_linux_stat, "Nproc") ;
rb_define_module_function(_nproc, "count_cpu_for_pid", count_cpu_for_pid, 1) ;
}
Version data entries
7 entries across 7 versions & 1 rubygems