main
json 123 lines 3.97 KB
Raw
1 [
2 {
3 "test_name": "throughput_llama8B_tp1",
4 "environment_variables": {
5 "PT_HPU_LAZY_MODE": 1,
6 "VLLM_CONTIGUOUS_PA": 1,
7 "VLLM_DEFRAG": 1
8 },
9 "parameters": {
10 "model": "meta-llama/Meta-Llama-3.1-8B-Instruct",
11 "tensor_parallel_size": 1,
12 "load_format": "dummy",
13 "dataset_path": "./ShareGPT_V3_unfiltered_cleaned_split.json",
14 "num_prompts": 1000,
15 "backend": "vllm",
16 "max-model-len": 2048,
17 "max-num-seqs": 512,
18 "async-scheduling": ""
19 }
20 },
21 {
22 "test_name": "throughput_llama70B_tp4",
23 "environment_variables": {
24 "PT_HPU_LAZY_MODE": 1,
25 "PT_HPU_ENABLE_LAZY_COLLECTIVES": 1,
26 "VLLM_CONTIGUOUS_PA": 1,
27 "VLLM_DEFRAG": 1
28 },
29 "parameters": {
30 "model": "meta-llama/Meta-Llama-3.1-70B-Instruct",
31 "tensor_parallel_size": 4,
32 "load_format": "dummy",
33 "dataset_path": "./ShareGPT_V3_unfiltered_cleaned_split.json",
34 "num_prompts": 1000,
35 "backend": "vllm",
36 "max-model-len": 2048,
37 "max-num-seqs": 512,
38 "async-scheduling": ""
39 }
40 },
41 {
42 "test_name": "throughput_mixtral8x7B_tp2",
43 "environment_variables": {
44 "PT_HPU_LAZY_MODE": 1,
45 "PT_HPU_ENABLE_LAZY_COLLECTIVES": 1,
46 "VLLM_CONTIGUOUS_PA": 1,
47 "VLLM_DEFRAG": 1
48 },
49 "parameters": {
50 "model": "mistralai/Mixtral-8x7B-Instruct-v0.1",
51 "tensor_parallel_size": 2,
52 "load_format": "dummy",
53 "dataset_path": "./ShareGPT_V3_unfiltered_cleaned_split.json",
54 "num_prompts": 1000,
55 "backend": "vllm",
56 "max-model-len": 2048,
57 "max-num-seqs": 512,
58 "async-scheduling": ""
59 }
60 },
61 {
62 "test_name": "throughput_deepseek_r1",
63 "environment_variables": {
64 "PT_HPU_LAZY_MODE": 1,
65 "PT_HPU_ENABLE_LAZY_COLLECTIVES": 1,
66 "VLLM_CONTIGUOUS_PA": 1,
67 "VLLM_DEFRAG": 1
68 },
69 "parameters": {
70 "model": "deepseek-ai/DeepSeek-R1",
71 "tensor_parallel_size": 8,
72 "load_format": "dummy",
73 "dataset_path": "./ShareGPT_V3_unfiltered_cleaned_split.json",
74 "dataset_name": "sharegpt",
75 "num_prompts": 1000,
76 "backend": "vllm",
77 "max-model-len": 2048,
78 "max-num-seqs": 384,
79 "async-scheduling": ""
80 }
81 },
82 {
83 "test_name": "throughput_llama4_maverick_17b128e_instruct_fp8",
84 "environment_variables": {
85 "PT_HPU_LAZY_MODE": 1,
86 "PT_HPU_ENABLE_LAZY_COLLECTIVES": 1,
87 "VLLM_CONTIGUOUS_PA": 1,
88 "VLLM_DEFRAG": 1
89 },
90 "parameters": {
91 "model": "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
92 "tensor_parallel_size": 8,
93 "dataset_path": "./ShareGPT_V3_unfiltered_cleaned_split.json",
94 "dataset_name": "sharegpt",
95 "num_prompts": 1000,
96 "backend": "vllm",
97 "max-model-len": 2048,
98 "max-num-seqs": 512,
99 "async-scheduling": "",
100 "enable_expert_parallel": ""
101 }
102 },
103 {
104 "test_name": "throughput_qwen3_8b",
105 "environment_variables": {
106 "PT_HPU_LAZY_MODE": 1,
107 "PT_HPU_ENABLE_LAZY_COLLECTIVES": 1,
108 "VLLM_CONTIGUOUS_PA": 1,
109 "VLLM_DEFRAG": 1
110 },
111 "parameters": {
112 "model": "Qwen/Qwen-3-8B",
113 "tensor_parallel_size": 1,
114 "load_format": "dummy",
115 "dataset_path": "./ShareGPT_V3_unfiltered_cleaned_split.json",
116 "dataset_name": "sharegpt",
117 "num_prompts": 1000,
118 "max-num-seqs": 512,
119 "backend": "vllm",
120 "async-scheduling": ""
121 }
122 }
123 ]