diff --git a/src/cpu/testers/traffic_gen/traffic_gen.cc b/src/cpu/testers/traffic_gen/traffic_gen.cc index 0f006e67e..b7f1ecd4e 100644 --- a/src/cpu/testers/traffic_gen/traffic_gen.cc +++ b/src/cpu/testers/traffic_gen/traffic_gen.cc @@ -119,6 +119,11 @@ TrafficGen::initState() unsigned int TrafficGen::drain(DrainManager *dm) { + if (!updateEvent.scheduled()) { + // no event has been scheduled yet (e.g. switched from atomic mode) + return 0; + } + if (retryPkt == NULL) { // shut things down nextPacketTick = MaxTick;