FM / utils

/usr/local/lib64/python3.6/site-packages/torch/utils/benchmark/utils/

/usr/local/lib64/python3.6/site-packages/torch/utils/benchmark/utils

Name	Size	Mode	Actions
valgrind_wrapper/	-	0755	rm
__pycache__/	-	0755	rm
common.py	13596	0644	edit dl rm
compare.py	12354	0644	edit dl rm
cpp_jit.py	6597	0644	edit dl rm
fuzzer.py	18211	0644	edit dl rm
sparse_fuzzer.py	5165	0644	edit dl rm
timeit_template.cpp	1017	0644	edit dl rm
timer.py	19292	0644	edit dl rm
_stubs.py	1142	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/utils/benchmark/utils/timeit_template.cpp (1017B)

/* C++ template for Timer.timeit

This template will be consumed by `cpp_jit.py`, and will replace:
    `GLOBAL_SETUP_TEMPLATE_LOCATION`,
    `SETUP_TEMPLATE_LOCATION`
      and
    `STMT_TEMPLATE_LOCATION`
sections with user provided statements.
*/
#include <chrono>

#include <c10/util/irange.h>
#include <pybind11/pybind11.h>
#include <c10/util/irange.h>
#include <torch/extension.h>

// Global setup. (e.g. #includes)
// GLOBAL_SETUP_TEMPLATE_LOCATION

double timeit(int n) {
    pybind11::gil_scoped_release no_gil;

// Setup
    // SETUP_TEMPLATE_LOCATION

{
        // Warmup
        // STMT_TEMPLATE_LOCATION
    }

// Main loop
    auto start_time = std::chrono::high_resolution_clock::now();
    for(const auto loop_idx : c10::irange(n)) {
        // STMT_TEMPLATE_LOCATION
    }
    auto end_time = std::chrono::high_resolution_clock::now();
    return std::chrono::duration<double>(end_time - start_time).count();
}

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
    m.def("timeit", &timeit);
}

/* C++ template for Timer.timeit This template will be consumed by `cpp_jit.py`, and will replace: `GLOBAL_SETUP_TEMPLATE_LOCATION`, `SETUP_TEMPLATE_LOCATION` and `STMT_TEMPLATE_LOCATION` sections with user provided statements. */ #include #include #include #include #include // Global setup. (e.g. #includes) // GLOBAL_SETUP_TEMPLATE_LOCATION double timeit(int n) { pybind11::gil_scoped_release no_gil; // Setup // SETUP_TEMPLATE_LOCATION { // Warmup // STMT_TEMPLATE_LOCATION } // Main loop auto start_time = std::chrono::high_resolution_clock::now(); for(const auto loop_idx : c10::irange(n)) { // STMT_TEMPLATE_LOCATION } auto end_time = std::chrono::high_resolution_clock::now(); return std::chrono::duration(end_time - start_time).count(); } PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { m.def("timeit", &timeit); }