[console_scripts]
torch_bench_gpu = torch_bench_gpu:main
