NVIDIA/nvbench

Add example showing how to parse additional arguments

開放

#87 建立於 2022年5月10日

 (6 則留言) (0 個反應) (0 位負責人)Cuda (115 個分叉)auto 404
P2: nice to havearea: examplesgood first issuetype: enhancement

倉庫指標

星標
 (914 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

Several users have asked about how to add custom arguments to their benchmarks (e.g. #86).

This is done by implementing an application specific main(argc, argv) function and linking to the nvbench::nvbench CMake target instead of nvbench::main. The custom main will need to parse out additional arguments, and then use the macros defined in main.cuh to parse the remaining args and run NVBench. Something like:

int main(int argc, char** argv)
{
  // parse custom args and remove them from (or rebuild) argc/argv here
  
  NVBENCH_MAIN_BODY(argc, argv);
}

A singleton or some global state is then needed to communicate the custom options to the benchmarks.

We should add an example that shows how to do this.

貢獻者指南