Skip to content

Commit d08a396

Browse files
Merge pull request #1092 from sstsimulator/devel
Automatically Merged using SST Master Branch Merger
2 parents 4e78e91 + 0cfc5e8 commit d08a396

File tree

2 files changed

+11
-6
lines changed

2 files changed

+11
-6
lines changed

src/sst/elements/ember/shmem/motifs/emberShmemGetNBI.h

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -58,13 +58,12 @@ class EmberShmemGetNBIGenerator : public EmberShmemGenerator {
5858
}
5959
assert( 2 == m_num_pes );
6060
m_other_pe = (m_my_pe + 1) % m_num_pes;
61-
enQ_malloc( evQ, &m_src, m_nelems * sizeof(TYPE) * 2);
61+
enQ_malloc( evQ, &m_src, m_nelems * sizeof(TYPE));
62+
enQ_malloc( evQ, &m_dest, m_nelems * sizeof(TYPE));
6263
break;
6364

6465
case 2:
6566

66-
m_dest = m_src.offset<TYPE>(m_nelems );
67-
6867
for ( int i = 0; i < m_nelems; i++ ) {
6968
m_src.at<TYPE>(i) = m_my_pe + i;
7069
}
@@ -73,8 +72,9 @@ class EmberShmemGetNBIGenerator : public EmberShmemGenerator {
7372

7473
enQ_barrier_all( evQ );
7574
enQ_getTime( evQ, &m_startTime );
76-
if ( m_my_pe == 1 ) {
75+
if ( m_my_pe == 0 ) {
7776
++m_phase;
77+
enQ_barrier_all( evQ );
7878
}
7979
break;
8080

@@ -86,12 +86,13 @@ class EmberShmemGetNBIGenerator : public EmberShmemGenerator {
8686
} else {
8787
enQ_quiet( evQ );
8888
enQ_getTime( evQ, &m_stopTime );
89+
enQ_barrier_all( evQ );
8990
}
9091
break;
9192

9293
case 4:
9394
ret = true;
94-
if ( 0 == m_my_pe ) {
95+
if ( m_my_pe != 0 ) {
9596
double time = m_stopTime-m_startTime;
9697
printf("%d:%s: count=%d, %.3lf ns \n",m_my_pe, getMotifName().c_str(), m_count,time/(double)m_count);
9798
}

src/sst/elements/ember/test/loadInfo.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,6 @@ def readWorkList(self, workList ):
293293

294294
ranksPerNode = self.numCores
295295
nidList = []
296-
297296
while len(cmdList):
298297
if "-" != cmdList[0][0]:
299298
break
@@ -329,6 +328,11 @@ def readWorkList(self, workList ):
329328
def parseCmd(self, motifPrefix, motifSuffix, cmdList, cmdNum ):
330329
motif = {}
331330

331+
tmp = cmdList[0].split('.')
332+
if len(tmp) == 2:
333+
motifPrefix = tmp[0] + '.'
334+
cmdList[0] = tmp[1]
335+
332336
tmp = 'motif' + str(cmdNum) + '.name'
333337
motif[ tmp ] = motifPrefix + cmdList[0] + motifSuffix
334338
cmdList.pop(0)

0 commit comments

Comments
 (0)