[LLM] Fix performance tests (#9596)
* Fix missing key for cpu_embedding * Remove 512 as it stuck for now * Small fix
This commit is contained in:
parent
8c8a27ded7
commit
1012507a40
6 changed files with 10 additions and 5 deletions
|
|
@ -26,4 +26,4 @@ in_out_pairs:
|
|||
- '2048-256'
|
||||
test_api:
|
||||
- "transformer_int4_gpu" # on Intel GPU
|
||||
|
||||
cpu_embedding: False # whether put embedding to CPU (only avaiable now for gpu win related test_api)
|
||||
|
|
|
|||
|
|
@ -11,4 +11,5 @@ in_out_pairs:
|
|||
- '1024-128'
|
||||
- '2048-256'
|
||||
test_api:
|
||||
- "transformer_int4_gpu" # on Intel GPU
|
||||
- "transformer_int4_gpu" # on Intel GPU
|
||||
cpu_embedding: False # whether put embedding to CPU (only avaiable now for gpu win related test_api)
|
||||
|
|
|
|||
|
|
@ -25,4 +25,5 @@ test_api:
|
|||
# - "transformer_int4_gpu" # on Intel GPU
|
||||
# - "optimize_model_gpu" # on Intel GPU
|
||||
# - "deepspeed_transformer_int4_cpu" # on Intel SPR Server
|
||||
|
||||
# - "transformer_int4_gpu_win" # on Intel GPU for Windows (catch GPU peak memory)
|
||||
cpu_embedding: False # whether put embedding to CPU (only avaiable now for gpu win related test_api)
|
||||
|
|
|
|||
|
|
@ -16,3 +16,6 @@ test_api:
|
|||
# - "ipex_fp16_gpu" # on Intel GPU
|
||||
# - "transformer_int4_gpu" # on Intel GPU
|
||||
# - "optimize_model_gpu" # on Intel GPU
|
||||
# - "deepspeed_transformer_int4_cpu" # on Intel SPR Server
|
||||
# - "transformer_int4_gpu_win" # on Intel GPU for Windows (catch GPU peak memory)
|
||||
cpu_embedding: False # whether put embedding to CPU (only avaiable now for gpu win related test_api)
|
||||
|
|
|
|||
|
|
@ -7,7 +7,7 @@ num_beams: 1 # default to greedy search
|
|||
low_bit: 'sym_int4' # default to use 'sym_int4' (i.e. symmetric int4)
|
||||
in_out_pairs:
|
||||
- '32-32'
|
||||
- '512-64'
|
||||
# - '512-64'
|
||||
# - '1024-128'
|
||||
test_api:
|
||||
# - "transformer_int4"
|
||||
|
|
|
|||
|
|
@ -18,7 +18,7 @@ num_beams: 1 # default to greedy search
|
|||
low_bit: 'sym_int4' # default to use 'sym_int4' (i.e. symmetric int4)
|
||||
in_out_pairs:
|
||||
- '32-32'
|
||||
- '512-64'
|
||||
# - '512-64'
|
||||
# - '1024-128'
|
||||
test_api:
|
||||
# - "transformer_int4"
|
||||
|
|
|
|||
Loading…
Reference in a new issue