example benchmark

Author: clay-arras

CMakeLists.txt

@@ -23,6 +23,7 @@ target_include_directories(pytokenizer_compile_config
 add_subdirectory(csrc/src/core)
 add_subdirectory(csrc/bindings)
 add_subdirectory(test)
+add_subdirectory(benchmarks)
 
 add_library(pytokenizer INTERFACE)
 target_link_libraries(pytokenizer INTERFACE pytokenizer_core)

benchmarks/CMakeLists.txt

@@ -0,0 +1,20 @@
+project(benchmarks LANGUAGES CXX)
+
+# FETCH CONTENT ---------------------------------
+include(FetchContent)
+set(FETCHCONTENT_TRY_FIND_PACKAGE TRUE)
+
+FetchContent_Declare(
+  benchmark
+  GIT_REPOSITORY https://github.com/google/benchmark.git
+  GIT_TAG        v1.9.0
+  GIT_SHALLOW    TRUE
+  FIND_PACKAGE_ARGS CONFIG
+)
+FetchContent_MakeAvailable(benchmark)
+
+# EXECUTABLES ---------------------------------
+add_executable(bench bench.cpp)
+target_link_libraries(bench
+  PRIVATE pytokenizer benchmark::benchmark
+)

benchmarks/bench.cpp

@@ -0,0 +1,21 @@
+#include <benchmark/benchmark.h>
+#include "libtokenizer/libtokenizer.hpp"
+
+void register_bench(const std::string& name) {
+  benchmark::RegisterBenchmark(
+      name,
+      [](benchmark::State& state) {
+        for (auto _ : state) {
+          pytokenizer::core::Tokenizer tokenizer;
+          tokenizer.foo();
+        }
+      });
+}
+
+int main(int argc, char** argv) {
+  benchmark::Initialize(&argc, argv);
+  register_bench("foo");
+
+  benchmark::RunSpecifiedBenchmarks();
+  return 0;
+}

csrc/include/libtokenizer/core/tokenizer.hpp

@@ -1,11 +1,13 @@
 #pragma once
 
+#include <string>
+
 namespace pytokenizer {
 namespace core {
 
 class Tokenizer {
 public:
-    void foo();
+    std::string foo();
 };
 
 } // namespace core

csrc/src/core/tokenizer.cpp

@@ -4,7 +4,7 @@
 namespace pytokenizer {
 namespace core {
 
-void Tokenizer::foo() { std::cout << "foo" << std::endl; }
+std::string Tokenizer::foo() { return "foo"; }
 
 } // namespace core
 } // namespace pytokenizer

test/playground.cpp

@@ -6,6 +6,6 @@
 
 int main() {
     pytokenizer::core::Tokenizer tokenizer;
-    tokenizer.foo();
+    std::cout << tokenizer.foo() << std::endl;
     return 0;
 }