derek/gem5
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

base,python: Added PcCountPair type and parameter

Browse Source 
This commit introduces a PcCountPair type that stores a Program Counter
address and an integer of counts for the Program Counter address.
The PcCountPair can be used in the same way and hashable in both C++
and Python.

Change-Id: I66d93e2c6a1d286cb9dd795ba97f8d887f67d503
Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/67193
Tested-by: kokoro <noreply+kokoro@google.com>
Reviewed-by: Bobby Bruce <bbruce@ucdavis.edu>
Maintainer: Bobby Bruce <bbruce@ucdavis.edu>
This commit is contained in:
Zhantong Qiu
2023-01-06 16:11:57 -08:00
parent e1601954f0
commit 717d3b239c
3 changed files with 167 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/python/pybind11/core.cc
View File

@@ -58,6 +58,7 @@
#include "sim/drain.hh"
#include "sim/serialize.hh"
#include "sim/sim_object.hh"
#include "cpu/probes/pc_count_pair.hh"
namespace py = pybind11;
@@ -163,6 +164,31 @@ init_range(py::module_ &m_native)
m.def("RangeSize", &RangeSize);
}
static void
init_pc(py::module_ &m_native)
{
py::module_ m = m_native.def_submodule("pc");
py::class_<PcCountPair>(m, "PcCountPair")
.def(py::init<>())
.def(py::init<Addr, int>())
.def("__eq__", [](const PcCountPair& self, py::object other) {
py::int_ pyPC = other.attr("get_pc")();
py::int_ pyCount = other.attr("get_count")();
uint64_t cPC = pyPC.cast<uint64_t>();
int cCount = pyCount.cast<int>();
return (self.getPC() == cPC && self.getCount() == cCount);
})
.def("__hash__", [](const PcCountPair& self){
py::int_ pyPC = py::cast(self.getPC());
py::int_ pyCount = py::cast(self.getCount());
return py::hash(py::make_tuple(pyPC, pyCount));
})
.def("__str__", &PcCountPair::to_string)
.def("get_pc", &PcCountPair::getPC)
.def("get_count", &PcCountPair::getCount)
;
}
static void
init_net(py::module_ &m_native)
{
@@ -307,6 +333,7 @@ pybind_init_core(py::module_ &m_native)
init_range(m_native);
init_net(m_native);
init_loader(m_native);
init_pc(m_native);
}
} // namespace gem5