feat(//cpp/benchmark): Increased workspace size for benchmark, may help

results and increase compatibility

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

Diff for: cpp/benchmark/main.cpp

@@ -136,6 +136,7 @@ int main(int argc, const char* argv[]) {
 
 #ifdef TRT
     auto extra_info = trtorch::ExtraInfo(dims);
+    extra_info.workspace_size = 1 << 24;
 
 #ifdef HALF
     extra_info.op_precision = at::kHalf;