cpu: implements vector registers
[gem5.git] / src / sim / pseudo_inst.cc
index aae2f60216d8c4d66cd811bb24c1b56e59864863..80737003c4e6de7e25cdd763d81c1f69651741aa 100644 (file)
@@ -1,4 +1,17 @@
 /*
+ * Copyright (c) 2010-2012 ARM Limited
+ * All rights reserved
+ *
+ * The license below extends only to copyright in the software and shall
+ * not be construed as granting a license to any other intellectual
+ * property including but not limited to intellectual property relating
+ * to a hardware implementation of the functionality of the software
+ * licensed hereunder.  You may use the software subject to the license
+ * terms below provided that you ensure that this notice is replicated
+ * unmodified and in its entirety in all distributions of the software,
+ * modified or unmodified, in source code or in binary form.
+ *
+ * Copyright (c) 2011 Advanced Micro Devices, Inc.
  * Copyright (c) 2003-2006 The Regents of The University of Michigan
  * All rights reserved.
  *
  * Authors: Nathan Binkert
  */
 
-#include <errno.h>
 #include <fcntl.h>
 #include <unistd.h>
 
+#include <cerrno>
+#include <fstream>
 #include <string>
+#include <vector>
 
-#include "sim/pseudo_inst.hh"
+#include "arch/kernel_stats.hh"
+#include "arch/utility.hh"
 #include "arch/vtophys.hh"
-#include "base/annotate.hh"
+#include "arch/pseudo_inst.hh"
+#include "base/debug.hh"
+#include "base/output.hh"
+#include "config/the_isa.hh"
 #include "cpu/base.hh"
-#include "cpu/thread_context.hh"
 #include "cpu/quiesce_event.hh"
-#include "kern/kernel_stats.hh"
-#include "sim/param.hh"
+#include "cpu/thread_context.hh"
+#include "debug/Loader.hh"
+#include "debug/PseudoInst.hh"
+#include "debug/Quiesce.hh"
+#include "debug/WorkItems.hh"
+#include "params/BaseCPU.hh"
+#include "sim/full_system.hh"
+#include "sim/process.hh"
+#include "sim/pseudo_inst.hh"
 #include "sim/serialize.hh"
+#include "sim/sim_events.hh"
 #include "sim/sim_exit.hh"
 #include "sim/stat_control.hh"
 #include "sim/stats.hh"
 #include "sim/system.hh"
-#include "sim/debug.hh"
 #include "sim/vptr.hh"
 
 using namespace std;
@@ -55,293 +80,641 @@ using namespace std;
 using namespace Stats;
 using namespace TheISA;
 
-namespace AlphaPseudo
+namespace PseudoInst {
+
+static inline void
+panicFsOnlyPseudoInst(const char *name)
+{
+    panic("Pseudo inst \"%s\" is only available in Full System mode.");
+}
+
+uint64_t
+pseudoInst(ThreadContext *tc, uint8_t func, uint8_t subfunc)
 {
-    bool doStatisticsInsts;
-    bool doCheckpointInsts;
-    bool doQuiesce;
-
-    void
-    arm(ThreadContext *tc)
-    {
-        if (tc->getKernelStats())
-            tc->getKernelStats()->arm();
+    uint64_t args[4];
+
+    DPRINTF(PseudoInst, "PseudoInst::pseudoInst(%i, %i)\n", func, subfunc);
+
+    // We need to do this in a slightly convoluted way since
+    // getArgument() might have side-effects on arg_num. We could have
+    // used the Argument class, but due to the possible side effects
+    // from getArgument, it'd most likely break.
+    int arg_num(0);
+    for (int i = 0; i < sizeof(args) / sizeof(*args); ++i) {
+        args[arg_num] = getArgument(tc, arg_num, sizeof(uint64_t), false);
+        ++arg_num;
     }
 
-    void
-    quiesce(ThreadContext *tc)
-    {
-        if (!doQuiesce)
-            return;
+    switch (func) {
+      case 0x00: // arm_func
+        arm(tc);
+        break;
 
-        tc->suspend();
-        if (tc->getKernelStats())
-            tc->getKernelStats()->quiesce();
-    }
+      case 0x01: // quiesce_func
+        quiesce(tc);
+        break;
 
-    void
-    quiesceNs(ThreadContext *tc, uint64_t ns)
-    {
-        if (!doQuiesce || ns == 0)
-            return;
+      case 0x02: // quiescens_func
+        quiesceSkip(tc);
+        break;
 
-        EndQuiesceEvent *quiesceEvent = tc->getQuiesceEvent();
+      case 0x03: // quiescecycle_func
+        quiesceNs(tc, args[0]);
+        break;
 
-        if (quiesceEvent->scheduled())
-            quiesceEvent->reschedule(curTick + Clock::Int::ns * ns);
-        else
-            quiesceEvent->schedule(curTick + Clock::Int::ns * ns);
+      case 0x04: // quiescetime_func
+        return quiesceTime(tc);
 
-        tc->suspend();
-        if (tc->getKernelStats())
-            tc->getKernelStats()->quiesce();
-    }
+      case 0x07: // rpns_func
+        return rpns(tc);
 
-    void
-    quiesceCycles(ThreadContext *tc, uint64_t cycles)
-    {
-        if (!doQuiesce || cycles == 0)
-            return;
+      case 0x09: // wakecpu_func
+        wakeCPU(tc, args[0]);
+        break;
 
-        EndQuiesceEvent *quiesceEvent = tc->getQuiesceEvent();
+      case 0x21: // exit_func
+        m5exit(tc, args[0]);
+        break;
 
-        if (quiesceEvent->scheduled())
-            quiesceEvent->reschedule(curTick +
-                                     tc->getCpuPtr()->cycles(cycles));
-        else
-            quiesceEvent->schedule(curTick +
-                                   tc->getCpuPtr()->cycles(cycles));
+      case 0x22:
+        m5fail(tc, args[0], args[1]);
+        break;
 
-        tc->suspend();
-        if (tc->getKernelStats())
-            tc->getKernelStats()->quiesce();
-    }
+      case 0x30: // initparam_func
+        return initParam(tc);
 
-    uint64_t
-    quiesceTime(ThreadContext *tc)
-    {
-        return (tc->readLastActivate() - tc->readLastSuspend()) / Clock::Int::ns;
-    }
+      case 0x31: // loadsymbol_func
+        loadsymbol(tc);
+        break;
 
-    void
-    ivlb(ThreadContext *tc)
-    {
-        if (tc->getKernelStats())
-            tc->getKernelStats()->ivlb();
-    }
+      case 0x40: // resetstats_func
+        resetstats(tc, args[0], args[1]);
+        break;
 
-    void
-    ivle(ThreadContext *tc)
-    {
-    }
+      case 0x41: // dumpstats_func
+        dumpstats(tc, args[0], args[1]);
+        break;
 
-    void
-    m5exit_old(ThreadContext *tc)
-    {
-        exitSimLoop(curTick, "m5_exit_old instruction encountered");
-    }
+      case 0x42: // dumprststats_func
+        dumpresetstats(tc, args[0], args[1]);
+        break;
 
-    void
-    m5exit(ThreadContext *tc, Tick delay)
-    {
-        Tick when = curTick + delay * Clock::Int::ns;
-        exitSimLoop(when, "m5_exit instruction encountered");
-    }
+      case 0x43: // ckpt_func
+        m5checkpoint(tc, args[0], args[1]);
+        break;
 
-    void
-    loadsymbol(ExecContext *xc)
-    {
-        const string &filename = xc->getCpuPtr()->system->params()->symbolfile;
-        if (filename.empty()) {
-            return;
-        }
+      case 0x4f: // writefile_func
+        return writefile(tc, args[0], args[1], args[2], args[3]);
 
-        std::string buffer;
-        ifstream file(filename.c_str());
+      case 0x50: // readfile_func
+        return readfile(tc, args[0], args[1], args[2]);
 
-        if (!file)
-            fatal("file error: Can't open symbol table file %s\n", filename);
+      case 0x51: // debugbreak_func
+        debugbreak(tc);
+        break;
 
-        while (!file.eof()) {
-            getline(file, buffer);
+      case 0x52: // switchcpu_func
+        switchcpu(tc);
+        break;
 
-            if (buffer.empty())
-                continue;
+      case 0x53: // addsymbol_func
+        addsymbol(tc, args[0], args[1]);
+        break;
 
-            int idx = buffer.find(' ');
-            if (idx == string::npos)
-                continue;
+      case 0x54: // panic_func
+        panic("M5 panic instruction called at %s\n", tc->pcState());
 
-            string address = "0x" + buffer.substr(0, idx);
-            eat_white(address);
-            if (address.empty())
-                continue;
+      case 0x5a: // work_begin_func
+        workbegin(tc, args[0], args[1]);
+        break;
 
-            // Skip over letter and space
-            string symbol = buffer.substr(idx + 3);
-            eat_white(symbol);
-            if (symbol.empty())
-                continue;
+      case 0x5b: // work_end_func
+        workend(tc, args[0], args[1]);
+        break;
 
-            Addr addr;
-            if (!to_number(address, addr))
-                continue;
+      case 0x55: // annotate_func
+      case 0x56: // reserved2_func
+      case 0x57: // reserved3_func
+      case 0x58: // reserved4_func
+      case 0x59: // reserved5_func
+        warn("Unimplemented m5 op (0x%x)\n", func);
+        break;
 
-            if (!xc->getSystemPtr()->kernelSymtab->insert(addr, symbol))
-                continue;
+      /* SE mode functions */
+      case 0x60: // syscall_func
+        m5Syscall(tc);
+        break;
 
+      case 0x61: // pagefault_func
+        m5PageFault(tc);
+        break;
 
-            DPRINTF(Loader, "Loaded symbol: %s @ %#llx\n", symbol, addr);
-        }
-        file.close();
+      default:
+        warn("Unhandled m5 op: 0x%x\n", func);
+        break;
     }
 
-    void
-    resetstats(ThreadContext *tc, Tick delay, Tick period)
-    {
-        if (!doStatisticsInsts)
-            return;
+    return 0;
+}
+
+void
+arm(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::arm()\n");
+    if (!FullSystem)
+        panicFsOnlyPseudoInst("arm");
+
+    if (tc->getKernelStats())
+        tc->getKernelStats()->arm();
+}
+
+void
+quiesce(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::quiesce()\n");
+    if (!FullSystem)
+        panicFsOnlyPseudoInst("quiesce");
+
+    if (!tc->getCpuPtr()->params()->do_quiesce)
+        return;
+
+    DPRINTF(Quiesce, "%s: quiesce()\n", tc->getCpuPtr()->name());
+
+    tc->suspend();
+    if (tc->getKernelStats())
+        tc->getKernelStats()->quiesce();
+}
+
+void
+quiesceSkip(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::quiesceSkip()\n");
+    if (!FullSystem)
+        panicFsOnlyPseudoInst("quiesceSkip");
+
+    BaseCPU *cpu = tc->getCpuPtr();
+
+    if (!cpu->params()->do_quiesce)
+        return;
+
+    EndQuiesceEvent *quiesceEvent = tc->getQuiesceEvent();
 
+    Tick resume = curTick() + 1;
 
-        Tick when = curTick + delay * Clock::Int::ns;
-        Tick repeat = period * Clock::Int::ns;
+    cpu->reschedule(quiesceEvent, resume, true);
 
-        using namespace Stats;
-        SetupEvent(Reset, when, repeat);
+    DPRINTF(Quiesce, "%s: quiesceSkip() until %d\n",
+            cpu->name(), resume);
+
+    tc->suspend();
+    if (tc->getKernelStats())
+        tc->getKernelStats()->quiesce();
+}
+
+void
+quiesceNs(ThreadContext *tc, uint64_t ns)
+{
+    DPRINTF(PseudoInst, "PseudoInst::quiesceNs(%i)\n", ns);
+    if (!FullSystem)
+        panicFsOnlyPseudoInst("quiesceNs");
+
+    BaseCPU *cpu = tc->getCpuPtr();
+
+    if (!cpu->params()->do_quiesce || ns == 0)
+        return;
+
+    EndQuiesceEvent *quiesceEvent = tc->getQuiesceEvent();
+
+    Tick resume = curTick() + SimClock::Int::ns * ns;
+
+    cpu->reschedule(quiesceEvent, resume, true);
+
+    DPRINTF(Quiesce, "%s: quiesceNs(%d) until %d\n",
+            cpu->name(), ns, resume);
+
+    tc->suspend();
+    if (tc->getKernelStats())
+        tc->getKernelStats()->quiesce();
+}
+
+void
+quiesceCycles(ThreadContext *tc, uint64_t cycles)
+{
+    DPRINTF(PseudoInst, "PseudoInst::quiesceCycles(%i)\n", cycles);
+    if (!FullSystem)
+        panicFsOnlyPseudoInst("quiesceCycles");
+
+    BaseCPU *cpu = tc->getCpuPtr();
+
+    if (!cpu->params()->do_quiesce || cycles == 0)
+        return;
+
+    EndQuiesceEvent *quiesceEvent = tc->getQuiesceEvent();
+
+    Tick resume = cpu->clockEdge(Cycles(cycles));
+
+    cpu->reschedule(quiesceEvent, resume, true);
+
+    DPRINTF(Quiesce, "%s: quiesceCycles(%d) until %d\n",
+            cpu->name(), cycles, resume);
+
+    tc->suspend();
+    if (tc->getKernelStats())
+        tc->getKernelStats()->quiesce();
+}
+
+uint64_t
+quiesceTime(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::quiesceTime()\n");
+    if (!FullSystem) {
+        panicFsOnlyPseudoInst("quiesceTime");
+        return 0;
     }
 
-    void
-    dumpstats(ThreadContext *tc, Tick delay, Tick period)
-    {
-        if (!doStatisticsInsts)
-            return;
+    return (tc->readLastActivate() - tc->readLastSuspend()) /
+        SimClock::Int::ns;
+}
+
+uint64_t
+rpns(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::rpns()\n");
+    return curTick() / SimClock::Int::ns;
+}
+
+void
+wakeCPU(ThreadContext *tc, uint64_t cpuid)
+{
+    DPRINTF(PseudoInst, "PseudoInst::wakeCPU(%i)\n", cpuid);
+    System *sys = tc->getSystemPtr();
+    ThreadContext *other_tc = sys->threadContexts[cpuid];
+    if (other_tc->status() == ThreadContext::Suspended)
+        other_tc->activate();
+}
 
+void
+m5exit(ThreadContext *tc, Tick delay)
+{
+    DPRINTF(PseudoInst, "PseudoInst::m5exit(%i)\n", delay);
+    Tick when = curTick() + delay * SimClock::Int::ns;
+    exitSimLoop("m5_exit instruction encountered", 0, when, 0, true);
+}
 
-        Tick when = curTick + delay * Clock::Int::ns;
-        Tick repeat = period * Clock::Int::ns;
+void
+m5fail(ThreadContext *tc, Tick delay, uint64_t code)
+{
+    DPRINTF(PseudoInst, "PseudoInst::m5fail(%i, %i)\n", delay, code);
+    Tick when = curTick() + delay * SimClock::Int::ns;
+    exitSimLoop("m5_fail instruction encountered", code, when, 0, true);
+}
 
-        using namespace Stats;
-        SetupEvent(Dump, when, repeat);
+void
+loadsymbol(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::loadsymbol()\n");
+    if (!FullSystem)
+        panicFsOnlyPseudoInst("loadsymbol");
+
+    const string &filename = tc->getCpuPtr()->system->params()->symbolfile;
+    if (filename.empty()) {
+        return;
     }
 
-    void
-    addsymbol(ThreadContext *tc, Addr addr, Addr symbolAddr)
-    {
-        char symb[100];
-        CopyStringOut(tc, symb, symbolAddr, 100);
-        std::string symbol(symb);
+    std::string buffer;
+    ifstream file(filename.c_str());
+
+    if (!file)
+        fatal("file error: Can't open symbol table file %s\n", filename);
+
+    while (!file.eof()) {
+        getline(file, buffer);
+
+        if (buffer.empty())
+            continue;
+
+        string::size_type idx = buffer.find(' ');
+        if (idx == string::npos)
+            continue;
+
+        string address = "0x" + buffer.substr(0, idx);
+        eat_white(address);
+        if (address.empty())
+            continue;
+
+        // Skip over letter and space
+        string symbol = buffer.substr(idx + 3);
+        eat_white(symbol);
+        if (symbol.empty())
+            continue;
+
+        Addr addr;
+        if (!to_number(address, addr))
+            continue;
+
+        if (!tc->getSystemPtr()->kernelSymtab->insert(addr, symbol))
+            continue;
+
 
         DPRINTF(Loader, "Loaded symbol: %s @ %#llx\n", symbol, addr);
+    }
+    file.close();
+}
+
+void
+addsymbol(ThreadContext *tc, Addr addr, Addr symbolAddr)
+{
+    DPRINTF(PseudoInst, "PseudoInst::addsymbol(0x%x, 0x%x)\n",
+            addr, symbolAddr);
+    if (!FullSystem)
+        panicFsOnlyPseudoInst("addSymbol");
+
+    char symb[100];
+    CopyStringOut(tc, symb, symbolAddr, 100);
+    std::string symbol(symb);
+
+    DPRINTF(Loader, "Loaded symbol: %s @ %#llx\n", symbol, addr);
+
+    tc->getSystemPtr()->kernelSymtab->insert(addr,symbol);
+    debugSymbolTable->insert(addr,symbol);
+}
 
-        tc->getSystemPtr()->kernelSymtab->insert(addr,symbol);
+uint64_t
+initParam(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::initParam()\n");
+    if (!FullSystem) {
+        panicFsOnlyPseudoInst("initParam");
+        return 0;
     }
 
-    void
-    anBegin(ThreadContext *tc, uint64_t cur)
-    {
-        Annotate::annotations.add(tc->getSystemPtr(), 0, cur >> 32, cur &
-                0xFFFFFFFF, 0,0);
+    return tc->getCpuPtr()->system->init_param;
+}
+
+
+void
+resetstats(ThreadContext *tc, Tick delay, Tick period)
+{
+    DPRINTF(PseudoInst, "PseudoInst::resetstats(%i, %i)\n", delay, period);
+    if (!tc->getCpuPtr()->params()->do_statistics_insts)
+        return;
+
+
+    Tick when = curTick() + delay * SimClock::Int::ns;
+    Tick repeat = period * SimClock::Int::ns;
+
+    Stats::schedStatEvent(false, true, when, repeat);
+}
+
+void
+dumpstats(ThreadContext *tc, Tick delay, Tick period)
+{
+    DPRINTF(PseudoInst, "PseudoInst::dumpstats(%i, %i)\n", delay, period);
+    if (!tc->getCpuPtr()->params()->do_statistics_insts)
+        return;
+
+
+    Tick when = curTick() + delay * SimClock::Int::ns;
+    Tick repeat = period * SimClock::Int::ns;
+
+    Stats::schedStatEvent(true, false, when, repeat);
+}
+
+void
+dumpresetstats(ThreadContext *tc, Tick delay, Tick period)
+{
+    DPRINTF(PseudoInst, "PseudoInst::dumpresetstats(%i, %i)\n", delay, period);
+    if (!tc->getCpuPtr()->params()->do_statistics_insts)
+        return;
+
+
+    Tick when = curTick() + delay * SimClock::Int::ns;
+    Tick repeat = period * SimClock::Int::ns;
+
+    Stats::schedStatEvent(true, true, when, repeat);
+}
+
+void
+m5checkpoint(ThreadContext *tc, Tick delay, Tick period)
+{
+    DPRINTF(PseudoInst, "PseudoInst::m5checkpoint(%i, %i)\n", delay, period);
+    if (!tc->getCpuPtr()->params()->do_checkpoint_insts)
+        return;
+
+    Tick when = curTick() + delay * SimClock::Int::ns;
+    Tick repeat = period * SimClock::Int::ns;
+
+    exitSimLoop("checkpoint", 0, when, repeat);
+}
+
+uint64_t
+readfile(ThreadContext *tc, Addr vaddr, uint64_t len, uint64_t offset)
+{
+    DPRINTF(PseudoInst, "PseudoInst::readfile(0x%x, 0x%x, 0x%x)\n",
+            vaddr, len, offset);
+    if (!FullSystem) {
+        panicFsOnlyPseudoInst("readfile");
+        return 0;
     }
 
-    void
-    anWait(ThreadContext *tc, uint64_t cur, uint64_t wait)
-    {
-        Annotate::annotations.add(tc->getSystemPtr(), 0, cur >> 32, cur &
-                0xFFFFFFFF, wait >> 32, wait & 0xFFFFFFFF);
+    const string &file = tc->getSystemPtr()->params()->readfile;
+    if (file.empty()) {
+        return ULL(0);
     }
 
+    uint64_t result = 0;
 
-    void
-    dumpresetstats(ThreadContext *tc, Tick delay, Tick period)
-    {
-        if (!doStatisticsInsts)
-            return;
+    int fd = ::open(file.c_str(), O_RDONLY, 0);
+    if (fd < 0)
+        panic("could not open file %s\n", file);
 
+    if (::lseek(fd, offset, SEEK_SET) < 0)
+        panic("could not seek: %s", strerror(errno));
 
-        Tick when = curTick + delay * Clock::Int::ns;
-        Tick repeat = period * Clock::Int::ns;
+    char *buf = new char[len];
+    char *p = buf;
+    while (len > 0) {
+        int bytes = ::read(fd, p, len);
+        if (bytes <= 0)
+            break;
 
-        using namespace Stats;
-        SetupEvent(Dump|Reset, when, repeat);
+        p += bytes;
+        result += bytes;
+        len -= bytes;
     }
 
-    void
-    m5checkpoint(ThreadContext *tc, Tick delay, Tick period)
-    {
-        if (!doCheckpointInsts)
-            return;
-        exitSimLoop("checkpoint");
+    close(fd);
+    CopyIn(tc, vaddr, buf, result);
+    delete [] buf;
+    return result;
+}
+
+uint64_t
+writefile(ThreadContext *tc, Addr vaddr, uint64_t len, uint64_t offset,
+            Addr filename_addr)
+{
+    DPRINTF(PseudoInst, "PseudoInst::writefile(0x%x, 0x%x, 0x%x, 0x%x)\n",
+            vaddr, len, offset, filename_addr);
+    ostream *os;
+
+    // copy out target filename
+    char fn[100];
+    std::string filename;
+    CopyStringOut(tc, fn, filename_addr, 100);
+    filename = std::string(fn);
+
+    if (offset == 0) {
+        // create a new file (truncate)
+        os = simout.create(filename, true);
+    } else {
+        // do not truncate file if offset is non-zero
+        // (ios::in flag is required as well to keep the existing data
+        //  intact, otherwise existing data will be zeroed out.)
+        os = simout.openFile(simout.directory() + filename,
+                            ios::in | ios::out | ios::binary);
     }
+    if (!os)
+        panic("could not open file %s\n", filename);
 
-    uint64_t
-    readfile(ThreadContext *tc, Addr vaddr, uint64_t len, uint64_t offset)
-    {
-        const string &file = tc->getCpuPtr()->system->params()->readfile;
-        if (file.empty()) {
-            return ULL(0);
-        }
+    // seek to offset
+    os->seekp(offset);
 
-        uint64_t result = 0;
+    // copy out data and write to file
+    char *buf = new char[len];
+    CopyOut(tc, buf, vaddr, len);
+    os->write(buf, len);
+    if (os->fail() || os->bad())
+        panic("Error while doing writefile!\n");
 
-        int fd = ::open(file.c_str(), O_RDONLY, 0);
-        if (fd < 0)
-            panic("could not open file %s\n", file);
+    simout.close(os);
 
-        if (::lseek(fd, offset, SEEK_SET) < 0)
-            panic("could not seek: %s", strerror(errno));
+    delete [] buf;
 
-        char *buf = new char[len];
-        char *p = buf;
-        while (len > 0) {
-            int bytes = ::read(fd, p, len);
-            if (bytes <= 0)
-                break;
+    return len;
+}
+
+void
+debugbreak(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::debugbreak()\n");
+    Debug::breakpoint();
+}
 
-            p += bytes;
-            result += bytes;
-            len -= bytes;
+void
+switchcpu(ThreadContext *tc)
+{
+    DPRINTF(PseudoInst, "PseudoInst::switchcpu()\n");
+    exitSimLoop("switchcpu");
+}
+
+//
+// This function is executed when annotated work items begin.  Depending on 
+// what the user specified at the command line, the simulation may exit and/or
+// take a checkpoint when a certain work item begins.
+//
+void
+workbegin(ThreadContext *tc, uint64_t workid, uint64_t threadid)
+{
+    DPRINTF(PseudoInst, "PseudoInst::workbegin(%i, %i)\n", workid, threadid);
+    tc->getCpuPtr()->workItemBegin();
+    System *sys = tc->getSystemPtr();
+    const System::Params *params = sys->params();
+    sys->workItemBegin(threadid, workid);
+
+    DPRINTF(WorkItems, "Work Begin workid: %d, threadid %d\n", workid, 
+            threadid);
+
+    //
+    // If specified, determine if this is the specific work item the user
+    // identified
+    //
+    if (params->work_item_id == -1 || params->work_item_id == workid) {
+
+        uint64_t systemWorkBeginCount = sys->incWorkItemsBegin();
+        int cpuId = tc->getCpuPtr()->cpuId();
+
+        if (params->work_cpus_ckpt_count != 0 &&
+            sys->markWorkItem(cpuId) >= params->work_cpus_ckpt_count) {
+            //
+            // If active cpus equals checkpoint count, create checkpoint
+            //
+            exitSimLoop("checkpoint");
         }
 
-        close(fd);
-        CopyIn(tc, vaddr, buf, result);
-        delete [] buf;
-        return result;
-    }
+        if (systemWorkBeginCount == params->work_begin_ckpt_count) {
+            //
+            // Note: the string specified as the cause of the exit event must
+            // exactly equal "checkpoint" inorder to create a checkpoint
+            //
+            exitSimLoop("checkpoint");
+        }
 
-    class Context : public ParamContext
-    {
-      public:
-        Context(const string &section) : ParamContext(section) {}
-        void checkParams();
-    };
-
-    Context context("pseudo_inst");
-
-    Param<bool> __quiesce(&context, "quiesce",
-                          "enable quiesce instructions",
-                          true);
-    Param<bool> __statistics(&context, "statistics",
-                             "enable statistics pseudo instructions",
-                             true);
-    Param<bool> __checkpoint(&context, "checkpoint",
-                             "enable checkpoint pseudo instructions",
-                             true);
-
-    void
-    Context::checkParams()
-    {
-        doQuiesce = __quiesce;
-        doStatisticsInsts = __statistics;
-        doCheckpointInsts = __checkpoint;
-    }
+        if (systemWorkBeginCount == params->work_begin_exit_count) {
+            //
+            // If a certain number of work items started, exit simulation
+            //
+            exitSimLoop("work started count reach");
+        }
 
-    void debugbreak(ThreadContext *tc)
-    {
-        debug_break();
+        if (cpuId == params->work_begin_cpu_id_exit) {
+            //
+            // If work started on the cpu id specified, exit simulation
+            //
+            exitSimLoop("work started on specific cpu");
+        }
     }
+}
+
+//
+// This function is executed when annotated work items end.  Depending on 
+// what the user specified at the command line, the simulation may exit and/or
+// take a checkpoint when a certain work item ends.
+//
+void
+workend(ThreadContext *tc, uint64_t workid, uint64_t threadid)
+{
+    DPRINTF(PseudoInst, "PseudoInst::workend(%i, %i)\n", workid, threadid);
+    tc->getCpuPtr()->workItemEnd();
+    System *sys = tc->getSystemPtr();
+    const System::Params *params = sys->params();
+    sys->workItemEnd(threadid, workid);
+
+    DPRINTF(WorkItems, "Work End workid: %d, threadid %d\n", workid, threadid);
+
+    //
+    // If specified, determine if this is the specific work item the user
+    // identified
+    //
+    if (params->work_item_id == -1 || params->work_item_id == workid) {
+
+        uint64_t systemWorkEndCount = sys->incWorkItemsEnd();
+        int cpuId = tc->getCpuPtr()->cpuId();
+
+        if (params->work_cpus_ckpt_count != 0 &&
+            sys->markWorkItem(cpuId) >= params->work_cpus_ckpt_count) {
+            //
+            // If active cpus equals checkpoint count, create checkpoint
+            //
+            exitSimLoop("checkpoint");
+        }
 
-    void switchcpu(ThreadContext *tc)
-    {
-        exitSimLoop("switchcpu");
+        if (params->work_end_ckpt_count != 0 &&
+            systemWorkEndCount == params->work_end_ckpt_count) {
+            //
+            // If total work items completed equals checkpoint count, create
+            // checkpoint
+            //
+            exitSimLoop("checkpoint");
+        }
+
+        if (params->work_end_exit_count != 0 &&
+            systemWorkEndCount == params->work_end_exit_count) {
+            //
+            // If total work items completed equals exit count, exit simulation
+            //
+            exitSimLoop("work items exit count reached");
+        }
     }
 }
+
+} // namespace PseudoInst