1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
#!/usr/bin/perl
use strict;
use warnings;
use utf8;
my ($file, $mpiJobs) = @ARGV;
defined($file) or die "USAGE: $0 filename\n";
my %h;
my $firstNode;
open(FILE, "<", "$file") or die "$0: Cannot open $file : $!\n";
while (<FILE>) {
chomp;
next if (/^#/ or $_ eq "");
my $f = $_;
$firstNode = $f;
if (!defined($h{"$f"})) {
$h{"$f"} = 1;
} else {
++$h{"$f"};
}
}
close(FILE);
my $nodes = scalar(keys %h);
die "$0: Nodes $nodes must be a multiple of mpiJobs $mpiJobs\n" if ($mpiJobs % $nodes != 0);
die "$0: No nodes!\n" if ($nodes == 0 or !defined($firstNode));
my $ppn = $h{"$firstNode"};
my $repeat = $mpiJobs/$nodes;
my $firstcall = 1;
for (my $i = 0; $i < $repeat; ++$i) {
foreach my $key (sort keys %h) {
my $n = $h{"$key"};
next if ($n == 0);
if ($firstcall) {
$firstcall = 0;
} else {
print ",";
}
print "$key";
}
}
|
