gem5 v24.0.0.0
Loading...
Searching...
No Matches
gem5py_m5.cc
Go to the documentation of this file.
1/*
2 * Copyright (c) 2019 ARM Limited
3 * All rights reserved
4 *
5 * The license below extends only to copyright in the software and shall
6 * not be construed as granting a license to any other intellectual
7 * property including but not limited to intellectual property relating
8 * to a hardware implementation of the functionality of the software
9 * licensed hereunder. You may use the software subject to the license
10 * terms below provided that you ensure that this notice is replicated
11 * unmodified and in its entirety in all distributions of the software,
12 * modified or unmodified, in source code or in binary form.
13 *
14 * Redistribution and use in source and binary forms, with or without
15 * modification, are permitted provided that the following conditions are
16 * met: redistributions of source code must retain the above copyright
17 * notice, this list of conditions and the following disclaimer;
18 * redistributions in binary form must reproduce the above copyright
19 * notice, this list of conditions and the following disclaimer in the
20 * documentation and/or other materials provided with the distribution;
21 * neither the name of the copyright holders nor the names of its
22 * contributors may be used to endorse or promote products derived from
23 * this software without specific prior written permission.
24 *
25 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
26 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
27 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
28 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
29 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
30 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
31 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
32 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
35 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36 */
37
38#include <pybind11/embed.h>
39
40#include <cstdlib>
41#include <iostream>
42
43namespace py = pybind11;
44
45/*
46 * This wrapper program runs python scripts using the python interpretter which
47 * will be built into gem5. Its first argument is the script to run, and then
48 * all subsequent arguments are passed to the python script as its argv.
49 */
50
51int
52main(int argc, const char **argv)
53{
54 py::scoped_interpreter guard;
55
56 // Embedded python doesn't set up sys.argv, so we'll do that ourselves.
57 py::list py_argv;
58 auto sys = py::module::import("sys");
59 if (py::hasattr(sys, "argv")) {
60 // sys.argv already exists, so grab that.
61 py_argv = sys.attr("argv");
62 } else {
63 // sys.argv doesn't exist, so create it.
64 sys.add_object("argv", py_argv);
65 }
66
67 auto importer = py::module_::import("importer");
68 importer.attr("install")();
69
70 // Clear out argv just in case it has something in it.
71 py_argv.attr("clear")();
72
73 if (argc < 2) {
74 std::cerr << "Usage: gem5py SCRIPT [arg] ..." << std::endl;
75 std::exit(1);
76 }
77
78 // Fill it with our argvs.
79 for (int i = 1; i < argc; i++)
80 py_argv.append(argv[i]);
81
82 // Actually call the script.
83 py::eval_file(argv[1]);
84
85 return 0;
86}
int main()

Generated on Tue Jun 18 2024 16:24:05 for gem5 by doxygen 1.11.0