Remove atomic from profiling

[ROCm/rccl commit: 31bd4236f1]
Cette révision appartient à :
Wenkai Du
2021-09-08 14:20:32 -05:00
Parent 310d51056f
révision d75504e9dc
4 fichiers modifiés avec 60 ajouts et 24 suppressions
+7 -3
Voir le fichier
@@ -280,12 +280,12 @@ static_assert(sizeof(struct ncclChannel) == 0x80*sizeof(int), "ncclChannel must
#pragma pack(pop) /* restore original alignment from stack */
#ifdef ENABLE_PROFILING
struct ncclProf {
struct ncclProfElem {
union {
struct {
uint64_t total_cycle;
uint64_t wait_cycle[MAXCHANNELS]; // total wait cycle
uint64_t wait_recv_cycle[MAXCHANNELS]; // recv wait cycle
uint64_t wait_cycle; // total wait cycle
uint64_t wait_recv_cycle; // recv wait cycle
// primtive cycles
uint64_t send_cycle;
uint64_t directSend_cycle;
@@ -316,6 +316,10 @@ struct ncclProf {
int data[0x80];
};
};
struct ncclProf {
struct ncclProfElem elems[MAXCHANNELS];
};
#endif
#ifdef ENABLE_COLLTRACE