[Benchmark] Add cpu options to bench scripts (#3915)

This commit is contained in:
Zedong Peng authored and GitHub committed 2024-04-09 21:30:03 -07:00
1 parent 11dd6ebb89
commit c013d32c75
2 files changed
+4 -4

No files matched your search

+2 -2
View File
@@ -169,8 +169,8 @@ if __name__ == '__main__':
"--device",
type=str,
default="cuda",
choices=["cuda"],
help='device type for vLLM execution, supporting CUDA only currently.')
choices=["cuda", "cpu"],
help='device type for vLLM execution, supporting CUDA and CPU.')
parser.add_argument('--block-size',
type=int,
default=16,
+2 -2
View File
@@ -329,8 +329,8 @@ if __name__ == "__main__":
"--device",
type=str,
default="cuda",
choices=["cuda"],
help='device type for vLLM execution, supporting CUDA only currently.')
choices=["cuda", "cpu"],
help='device type for vLLM execution, supporting CUDA and CPU.')
parser.add_argument(
"--enable-prefix-caching",
action='store_true',