Skip to content

Commit feec95d

Browse files
author
Ted Themistokleous
committed
Only print quantizer info for int8 runs
1 parent 09814cf commit feec95d

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

quantization/nlp/bert/migraphx/e2e_migraphx_bert_example.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -435,10 +435,11 @@ def output_run_config(flags, samples):
435435
print ("Samples: " + str(samples) + " Batch size: " + str(flags.batch))
436436
print ("Sequence length: " + str(flags.seq_len))
437437
print ("Model Quantization: fp16:" + str(flags.fp16) + " int8:" + str(flags.int8))
438-
if flags.ort_quant and flags.int8:
439-
print ("Quantizer: Onnxruntime")
440-
else:
441-
print("Quantizer: MIGraphX")
438+
if flags.int8:
439+
if flags.ort_quant
440+
print ("Quantizer: Onnxruntime")
441+
else:
442+
print("Quantizer: MIGraphX")
442443

443444

444445
if __name__ == '__main__':

0 commit comments

Comments
 (0)