Skip to content

Commit

Permalink
set fp8 option as true
Browse files Browse the repository at this point in the history
  • Loading branch information
anandhu-eng committed Jul 8, 2024
1 parent 0b80cbc commit 0247b0a
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions script/app-mlperf-inference-nvidia/_cm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -1025,6 +1025,10 @@ variations:
default_variations:
batch-size: batch_size.1

llama2-70b_,run_harness:
env:
CM_MLPERF_NVIDIA_HARNESS_USE_FP8: 'True'

gptj_,run_harness:
deps:
- tags: install,pytorch,from.src
Expand Down

0 comments on commit 0247b0a

Please sign in to comment.