zbench/zbench.cpp at main · smarcink/zbench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#include <iostream>
#include <vector>

#include <pybind11/pybind11.h>
// #include <pybind11/stl.h>
// #include <pybind11/numpy.h>
// #include <pybind11_json/pybind11_json.hpp>
// namespace py = pybind11;
// #include <nlohmann/json.hpp>
// namespace nl = nlohmann;

#include "dev_test.h"
#include "rt_igdext.h"

PYBIND11_MODULE(zbench, m)
{
    m.doc() = "pybind11 testbed based on d3d12 for CM kernels";

    // dev function / feature test
    m.def("add", &add, "A function which adds two numbers", py::arg("i") = 1, py::arg("j") = 2);
    m.def("test_bind", &test_bind, py::arg("mode"), py::arg("input"),"test pybind11 for array transfering");
    m.def("test_set_json", &test_set_json, "pass py::object to a C++ function that takes an nlohmann::json");
    m.def("test_get_json", &test_get_json, "return py::object from a C++ function that returns an nlohmann::json");

    m.def("launch_rt_igdext", &launch_rt_igdext, "A function which adds two numbers",
          py::arg("cm_file") = "gemm_nchw_fp16.cpp",
          py::arg("build_options") = "None"
        );

//     // Deprecated sample bench on L0 runtime.
//     m.def("run_sgemm", &run_sgemm, "A function which adds two numbers",
//           py::arg("m") = 1024, py::arg("niterations") = 1,
//           py::arg("gy") = 1, py::arg("gx") = 4,
//           py::arg("bin_file") = "sgemm_genx.bin",
//           py::arg("fn_name") = "sgemm_kernel"
//           );

//     // DPAS bgemm cm kernel
//     m.def("run_bgemm", &run_bgemm, "A function which adds two numbers",
//           py::arg("M") = 128, py::arg("N") = 128,py::arg("K") = 128,
//           py::arg("threadWidth") = 4, py::arg("threadHeight") = 4,
//           py::arg("groupWidth") = 1, py::arg("groupHeight") = 1,
//           py::arg("bin_file") = "bgemm_dpas_genx.bin",
//           py::arg("fn_name") = "bgemm_dpas"
//           );

//     // code for testing the numpy array binding
//     m.def("run_kernel", &run_kernel, "A function which adds two numbers",
//           py::arg("bin_file") = "bgemm_dpas_genx.bin",
//           py::arg("spirv_file") = "bgemm_dpas_genx.spv",
//           py::arg("fn_name") = "bgemm_dpas"
//           );

//     m.def("run_gemm_nchw_fp16", &run_gemm_nchw_fp16, "A function which adds two numbers",
//           py::arg("bin_file") = "bgemm_dpas_genx.bin",
//           py::arg("spirv_file") = "bgemm_dpas_genx.spv",
//           py::arg("fn_name") = "bgemm_dpas"
//           );

}