Skip to content

Commit

Permalink
Do not skip llama_v2_7b_16h on NVIDIA A100 40G. (#2124)
Browse files Browse the repository at this point in the history
Summary:
This PR partially reverts #2095, since #2064 seems not to be an issue anymore.

Pull Request resolved: #2124

Reviewed By: suez1224

Differential Revision: D53093766

Pulled By: xuzhao9

fbshipit-source-id: 157a01dec22e48b5ee1cb1260070a6d270aec4f8
  • Loading branch information
ysiraichi authored and facebook-github-bot committed Jan 25, 2024
1 parent 236d6bf commit c6e2004
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions torchbenchmark/models/llama_v2_7b_16h/metadata.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,5 @@ eval_nograd: true
not_implemented:
- device: cpu
- device: NVIDIA A10G
# TODO: llama_v2_7b_16h accuracy test will cause "CUBLAS_STATUS_NOT_INITIALIZED" Error
# https://github.com/pytorch/benchmark/issues/2064
- device: NVIDIA A100-SXM4-40GB
test: eval
train_benchmark: false
train_deterministic: false

0 comments on commit c6e2004

Please sign in to comment.